Skip to content

GitLab

Commit 3c284567 authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

missing ALL_TASKS import

parent 483f86d9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
lm_eval/tasks/__init__.py
View file @ 3c284567
......@@ -12,6 +12,7 @@ from lm_eval.api.registry import (
register_group,
TASK_REGISTRY,
GROUP_REGISTRY,
ALL_TASKS,
)
......@@ -58,8 +59,6 @@ def include_task_folder(task_dir):
task_dir = os.path.dirname(os.path.abspath(__file__)) + "/"
include_task_folder(task_dir)
# ALL_TASKS = set(sorted(list(TASK_REGISTRY.keys()) + list(GROUP_REGISTRY.keys())))
def get_task(task_name, config):
try:
......
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