Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "4a91269a0ee5a5814687b20150b172e0d586e129"
Commit c2e32987 authored by Jon Tow's avatar Jon Tow
Browse files

Fix dataset value feature

parent 897ae76d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lm_eval/datasets/triviaqa/triviaqa.py
View file @ c2e32987
...@@ -67,7 +67,7 @@ class TriviaQa(datasets.GeneratorBasedBuilder): ...@@ -67,7 +67,7 @@ class TriviaQa(datasets.GeneratorBasedBuilder):
"aliases": datasets.features.Sequence( "aliases": datasets.features.Sequence(
datasets.Value("string"), datasets.Value("string"),
), ),
"values": datasets.Value("string") "value": datasets.Value("string")
} }
} }
) )
...@@ -110,6 +110,6 @@ class TriviaQa(datasets.GeneratorBasedBuilder): ...@@ -110,6 +110,6 @@ class TriviaQa(datasets.GeneratorBasedBuilder):
"question": data["Question"], "question": data["Question"],
"answer": { "answer": {
"aliases": data["Answer"]["Aliases"], "aliases": data["Answer"]["Aliases"],
"values": data["Answer"]["Value"], "value": data["Answer"]["Value"],
} }
} }
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