Skip to content

GitLab

Unverified Commit e35386d9 authored by Leo Gao's avatar Leo Gao Committed by GitHub
Browse files

Fix version issue

parent db85d5ae
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
lm_eval/tasks/lambada_multilingual.py
View file @ e35386d9
...@@ -40,5 +40,8 @@ class MultilingualLAMBADA(lambada.LAMBADA): ...@@ -40,5 +40,8 @@ class MultilingualLAMBADA(lambada.LAMBADA):
def construct_tasks(): def construct_tasks():
tasks = {} tasks = {}
for lang in LANGS: 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 return tasks
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment