minor fix

f5d0f8e6 · lintangsutawika · 4fe3409e · f5d0f8e6
Commit f5d0f8e6 authored Aug 23, 2023 by lintangsutawika
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/api/task.py lm_eval/api/task.py +1 -1

No files found.
--- a/lm_eval/api/task.py
+++ b/lm_eval/api/task.py
@@ -666,7 +666,7 @@ class ConfigurableTask(Task):
        if test_choice is not None:
            check_choices = test_choice
        else:
-            check_choices = test_target
+            check_choices = [test_target]

        for choice in check_choices:
            choice_has_whitespace = True if " " in choice else False