Skip to content

GitLab

Commit b2aee121 authored by lintangsutawika's avatar lintangsutawika
Browse files

add checks

parent 3317c627
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/api/task.py
View file @ b2aee121
...@@ -655,7 +655,7 @@ class ConfigurableTask(Task): ...@@ -655,7 +655,7 @@ class ConfigurableTask(Task):
if type(test_target) is list: if type(test_target) is list:
self.multiple_target = len(test_target) self.multiple_target = len(test_target)
else: else:
if type(test_target) is int: if (type(test_target) is int) and (test_choice is not None):
test_target = [self.doc_to_choice(test_target)[test_target]] test_target = [self.doc_to_choice(test_target)[test_target]]
else: else:
test_target = [test_target] test_target = [test_target]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment