Skip to content

GitLab

Commit 95908d8b authored by h-albert-lee's avatar h-albert-lee
Browse files

Modify code for flake8

parent 9221cb57
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lm_eval/tasks/kmmlu/utils.py
View file @ 95908d8b
import datasets
def process_docs(dataset: datasets.Dataset) -> datasets.Dataset:
def _process_doc(doc):
instruction = (
......@@ -12,9 +13,8 @@ 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
return dataset.map(_process_doc)
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