Fix version issue

e35386d9 · Leo Gao · GitHub · db85d5ae · e35386d9
Unverified Commit e35386d9 authored Jul 02, 2021 by Leo Gao Committed by GitHub Jul 02, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

lm_eval/tasks/lambada_multilingual.py lm_eval/tasks/lambada_multilingual.py +4 -1

No files found.
--- a/lm_eval/tasks/lambada_multilingual.py
+++ b/lm_eval/tasks/lambada_multilingual.py
@@ -40,5 +40,8 @@ class MultilingualLAMBADA(lambada.LAMBADA):
 def construct_tasks():
    tasks = {}
    for lang in LANGS:
-        tasks[f"lambada_mt_{lang}"] = partial(MultilingualLAMBADA, lang=lang)
+        class MultilingualLAMBADAInstance(MultilingualLAMBADA):
+            def __init__(self):
+              super().__init__(lang)
+        tasks[f"lambada_mt_{lang}"] = MultilingualLAMBADAInstance
    return tasks