Fix dataset value feature

c2e32987 · Jon Tow · 897ae76d · c2e32987
Commit c2e32987 authored Mar 18, 2022 by Jon Tow
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

lm_eval/datasets/triviaqa/triviaqa.py lm_eval/datasets/triviaqa/triviaqa.py +2 -2

No files found.
--- a/lm_eval/datasets/triviaqa/triviaqa.py
+++ b/lm_eval/datasets/triviaqa/triviaqa.py
@@ -67,7 +67,7 @@ class TriviaQa(datasets.GeneratorBasedBuilder):
                    "aliases":  datasets.features.Sequence(
                        datasets.Value("string"),
                    ),
-                    "values": datasets.Value("string")
+                    "value": datasets.Value("string")
                }
            }
        )
@@ -110,6 +110,6 @@ class TriviaQa(datasets.GeneratorBasedBuilder):
                    "question": data["Question"],
                    "answer": {
                        "aliases": data["Answer"]["Aliases"],
-                        "values": data["Answer"]["Value"],
+                        "value": data["Answer"]["Value"],
                    }
                }