fix #3837

2024-05-30 00:52:26 +08:00 · 2024-05-30 00:52:26 +08:00 · 3404e8f302
parent 483eb47e5d
commit 3404e8f302
1 changed files with 1 additions and 1 deletions
--- a/evaluation/mmlu/mmlu.py
+++ b/evaluation/mmlu/mmlu.py
@ -154,7 +154,7 @@ class MMLU(datasets.GeneratorBasedBuilder):
        ]

    def _generate_examples(self, filepath):
-        df = pd.read_csv(filepath)
+        df = pd.read_csv(filepath, header=None)
        df.columns = ["question", "A", "B", "C", "D", "answer"]

        for i, instance in enumerate(df.to_dict(orient="records")):