Commit 48d861ce authored by baberabb's avatar baberabb
Browse files

fixed fixture

parent ba75c838
...@@ -6,7 +6,7 @@ from lm_eval.api.task import ConfigurableTask ...@@ -6,7 +6,7 @@ from lm_eval.api.task import ConfigurableTask
# Using fixtures to get the task class and limit # Using fixtures to get the task class and limit
@pytest.fixture() @pytest.fixture()
def task_class(task_name: List[str]) -> ConfigurableTask: def task_class() -> ConfigurableTask:
task_name = ["arc_easy"] task_name = ["arc_easy"]
x = [cls for name, cls in tasks.TASK_REGISTRY.items() if name in task_name] x = [cls for name, cls in tasks.TASK_REGISTRY.items() if name in task_name]
return x[0] return x[0]
......
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