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
f7d98500
Commit
f7d98500
authored
Sep 12, 2023
by
lintangsutawika
Browse files
allow nested groups
parent
3839125a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
15 additions
and
4 deletions
+15
-4
lm_eval/tasks/__init__.py
lm_eval/tasks/__init__.py
+15
-4
No files found.
lm_eval/tasks/__init__.py
View file @
f7d98500
...
...
@@ -136,6 +136,9 @@ def get_task_dict(task_name_list: List[Union[str, dict, Task]], **kwargs):
task_name_from_config_dict
=
{}
task_name_from_object_dict
=
{}
if
type
(
task_name_list
)
!=
list
:
task_name_list
=
[
task_name_list
]
for
task_element
in
task_name_list
:
if
isinstance
(
task_element
,
str
):
...
...
@@ -143,12 +146,20 @@ def get_task_dict(task_name_list: List[Union[str, dict, Task]], **kwargs):
group_name
=
task_element
for
task_name
in
GROUP_REGISTRY
[
task_element
]:
if
task_name
not
in
task_name_from_registry_dict
:
task_obj
=
get_task_dict
(
task_name
)
if
task_name
in
task_obj
.
keys
():
task_dict
=
{
task_name
:
(
group_name
,
task_obj
[
task_name
]),
}
else
:
task_dict
=
{
task_name
:
(
group_name
,
None
),
**
task_obj
,
}
task_name_from_registry_dict
=
{
**
task_name_from_registry_dict
,
task_name
:
(
group_name
,
get_task
(
task_name
=
task_name
,
config
=
config
),
),
**
task_dict
,
}
else
:
task_name
=
task_element
...
...
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