From 22ea032f2e8c22b83a8f67008d05ef86c3251206 Mon Sep 17 00:00:00 2001 From: jon-tow Date: Thu, 28 Apr 2022 02:45:58 -0400 Subject: [PATCH] Fix `mlsum` task names after split update --- lm_eval/tasks/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lm_eval/tasks/__init__.py b/lm_eval/tasks/__init__.py index 135ade13e2..39af5d8f3d 100644 --- 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,