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

Add positional args to `gsm8k` `download_file`

Merge pull request #296 from jon-tow/gsm8k-fix
Add positional args to `gsm8k` `download_file`
fcaea0e3 · Leo Gao · GitHub · cec2fe23 · f91a70d6 · fcaea0e3
Unverified Commit fcaea0e3 authored Mar 06, 2022 by Leo Gao Committed by GitHub Mar 06, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

lm_eval/tasks/gsm8k.py lm_eval/tasks/gsm8k.py +2 -1

No files found.
--- a/lm_eval/tasks/gsm8k.py
+++ b/lm_eval/tasks/gsm8k.py
@@ -56,7 +56,8 @@ class GradeSchoolMath8K(Task):
        ]
        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