datasets/nsmc.py key error modified

0d94326e · cardy20 · 1bf3c505 · 0d94326e
Commit 0d94326e authored Apr 30, 2023 by cardy20
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/datasets/nsmc/nsmc.py lm_eval/datasets/nsmc/nsmc.py +1 -1

No files found.
--- a/lm_eval/datasets/nsmc/nsmc.py
+++ b/lm_eval/datasets/nsmc/nsmc.py
@@ -80,7 +80,7 @@ class NSMC(datasets.GeneratorBasedBuilder):
            datasets.SplitGenerator(
                name=datasets.Split.TEST,
                gen_kwargs={
-                    "filepath": downloaded_files["validation"],
+                    "filepath": downloaded_files["test"],
                    "split": "test",
                },
            ),