Merge pull request #296 from jon-tow/gsm8k-fix

Add positional args to `gsm8k` `download_file`
EleutherAI · Mar 6, 2022 · fcaea0e · fcaea0e
2 parents cec2fe2 + f91a70d
commit fcaea0e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lm_eval/tasks/gsm8k.py b/lm_eval/tasks/gsm8k.py
@@ -56,7 +56,8 @@ def download(self):
         ]
         for split in splits:
             file = self.DATASET_PATH / f"{split['name']}.jsonl"
-            download_file(f"{base_url}/{split['name']}.jsonl", str(file), split["checksum"])
+            url = f"{base_url}/{split['name']}.jsonl"
+            download_file(url, local_file=str(file), expected_checksum=split["checksum"])
 
     def has_training_docs(self):
         return True