fixed doc_to_choice in race

e9432fad · lintangsutawika · 9218bcff · e9432fad
Commit e9432fad authored Jul 13, 2023 by lintangsutawika
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/tasks/race/preprocess_race.py lm_eval/tasks/race/preprocess_race.py +1 -1

No files found.
--- a/lm_eval/tasks/race/preprocess_race.py
+++ b/lm_eval/tasks/race/preprocess_race.py
@@ -15,7 +15,7 @@ def get_answer_option(problem):
    return problem["options"][answer]
-def create_choices(doc):
+def doc_to_choice(doc):
    problem = last_problem(doc)
    choices = [problem["options"][i] for i in range(4)]
    return choices