Skip to content

GitLab

Commit 0905615f authored by lintangsutawika's avatar lintangsutawika
Browse files

reformat

parent 39c40277
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 18 deletions
lm_eval/tasks/__init__.py
View file @ 0905615f
...@@ -67,16 +67,12 @@ class TaskManager: ...@@ -67,16 +67,12 @@ class TaskManager:
return False return False
def _name_is_task(self, name) -> bool: def _name_is_task(self, name) -> bool:
if self._name_is_registered(name) and ( if self._name_is_registered(name) and (self.task_index[name]["type"] == "task"):
self.task_index[name]["type"] == "task"
):
return True return True
return False return False
def _name_is_tag(self, name) -> bool: def _name_is_tag(self, name) -> bool:
if self._name_is_registered(name) and ( if self._name_is_registered(name) and (self.task_index[name]["type"] == "tag"):
self.task_index[name]["type"] == "tag"
):
return True return True
return False return False
...@@ -175,12 +171,8 @@ class TaskManager: ...@@ -175,12 +171,8 @@ class TaskManager:
else: else:
# group_name = name_or_config # group_name = name_or_config
group_name = ConfigurableGroup( group_name = ConfigurableGroup(
config={ config={"group": name_or_config, "task": subtask_list}
"group": name_or_config, )
"task": subtask_list
}
)
if isinstance(name_or_config, dict): if isinstance(name_or_config, dict):
if update_config is not None: if update_config is not None:
...@@ -196,7 +188,9 @@ class TaskManager: ...@@ -196,7 +188,9 @@ class TaskManager:
if self._name_is_group(name) or self._name_is_tag(name): if self._name_is_group(name) or self._name_is_tag(name):
group_name = name group_name = name
update_config = { update_config = {
k: v for k, v in name_or_config.items() if k not in ["task", "group"] k: v
for k, v in name_or_config.items()
if k not in ["task", "group"]
} }
subtask_list = self._get_tasklist(name) subtask_list = self._get_tasklist(name)
if subtask_list == -1: if subtask_list == -1:
...@@ -206,11 +200,8 @@ class TaskManager: ...@@ -206,11 +200,8 @@ class TaskManager:
else: else:
# group_name = name # group_name = name
group_name = ConfigurableGroup( group_name = ConfigurableGroup(
config={ config={"group": group_name, "task": subtask_list}
"group": group_name, )
"task": subtask_list
}
)
else: else:
if self._name_is_registered(name): if self._name_is_registered(name):
base_task_config = self._get_config(name) base_task_config = self._get_config(name)
......
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