fix after changes

Lightning-AI · Apr 6, 2024 · 494af64 · 494af64
1 parent 7acd9a1
commit 494af64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litgpt/eval/evaluate.py b/litgpt/eval/evaluate.py
@@ -72,7 +72,7 @@ def convert_and_evaluate(
     if tasks is None:
         from lm_eval.tasks import TaskManager
         taskm = TaskManager()
-        print("\n".join(taskm.task_index.values())
+        print("\n".join(taskm.task_index.keys()))
         print(
             "\n\nTo evaluate multiple tasks, you can chain the task names "
             "listed above via a comma-separated list."