Merge pull request #15 from bigscience-workshop/update-mlsum-task-name

Fix `mlsum` task names after split update
OpenGPTX · Apr 28, 2022 · 9cd7023 · 9cd7023
2 parents efd319b + 22ea032
commit 9cd7023
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lm_eval/tasks/__init__.py b/lm_eval/tasks/__init__.py
@@ -290,8 +290,8 @@
     "blimp_wh_vs_that_with_gap_long_distance": blimp.BlimpWhVsThatWithGapLongDistance,
 
     #GEM/mlsum
-    "mlsum_es":gem_mlsum.MLSUMEs,
-    "mlsum_de":gem_mlsum.MLSUMDe,
+    "mlsum_es":gem_mlsum.GEMMLSUMEs,
+    "mlsum_de":gem_mlsum.GEMMLSUMDe,
     "mlsum_es_covid_challenge_set":gem_mlsum.GEMMLSUMEsChallgeTestCovid,
     "mlsum_de_covid_challenge_set":gem_mlsum.GEMMLSUMDeChallgeTestCovid,