Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
lm-evaluation-harness
Commits
51b795cf
Commit
51b795cf
authored
May 15, 2023
by
lintangsutawika
Browse files
adjustments to ALL_TASKS
parent
64d07df7
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
2 deletions
+4
-2
lm_eval/tasks/__init__.py
lm_eval/tasks/__init__.py
+4
-2
No files found.
lm_eval/tasks/__init__.py
View file @
51b795cf
...
@@ -75,7 +75,7 @@ for root, subdirs, file_list in os.walk(task_dir):
...
@@ -75,7 +75,7 @@ for root, subdirs, file_list in os.walk(task_dir):
TASK_REGISTRY
=
task_registry
TASK_REGISTRY
=
task_registry
GROUP_REGISTRY
=
group_registry
GROUP_REGISTRY
=
group_registry
ALL_TASKS
=
sorted
(
list
(
TASK_REGISTRY
))
ALL_TASKS
=
sorted
(
list
(
TASK_REGISTRY
.
keys
())
+
list
(
GROUP_REGISTRY
.
keys
()
))
def
get_task
(
task_name
,
config
):
def
get_task
(
task_name
,
config
):
try
:
try
:
...
@@ -105,7 +105,9 @@ def get_task_name_from_config(task_config):
...
@@ -105,7 +105,9 @@ def get_task_name_from_config(task_config):
# TODO: pass num_fewshot and other cmdline overrides in a better way
# TODO: pass num_fewshot and other cmdline overrides in a better way
def
get_task_dict
(
task_name_list
:
List
[
Union
[
str
,
dict
,
Task
]],
config
,
**
kwargs
):
def
get_task_dict
(
task_name_list
:
List
[
Union
[
str
,
dict
,
Task
]],
**
kwargs
):
config
=
{
**
kwargs
}
task_name_from_registry_dict
=
{}
task_name_from_registry_dict
=
{}
task_name_from_config_dict
=
{}
task_name_from_config_dict
=
{}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment