Skip to content

GitLab

Commit f91a70d6 authored by Jonathan Tow's avatar Jonathan Tow
Browse files

Add positional args to `gsm8k` `download_file`

parent cec2fe23
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
lm_eval/tasks/gsm8k.py
View file @ f91a70d6
...@@ -56,7 +56,8 @@ class GradeSchoolMath8K(Task): ...@@ -56,7 +56,8 @@ class GradeSchoolMath8K(Task):
] ]
for split in splits: for split in splits:
file = self.DATASET_PATH / f"{split['name']}.jsonl" 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): def has_training_docs(self):
return True return True
......
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