reformatted with Black

520d8ce2 · h-albert-lee · 95908d8b · 520d8ce2
Commit 520d8ce2 authored Dec 10, 2023 by h-albert-lee
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/tasks/kmmlu/utils.py lm_eval/tasks/kmmlu/utils.py +1 -1

No files found.
--- a/lm_eval/tasks/kmmlu/utils.py
+++ b/lm_eval/tasks/kmmlu/utils.py
@@ -13,7 +13,7 @@ def process_docs(dataset: datasets.Dataset) -> datasets.Dataset:
        out_doc = {
            "question": instruction,
            "choices": ["(1)", "(2)", "(3)", "(4)"],
-            "gold": int(doc["gold"]) - 1
+            "gold": int(doc["gold"]) - 1,
        }
        return out_doc