Commit 1b0858e7 authored by lintangsutawika's avatar lintangsutawika
Browse files

fixed bug

parent 8864e1a9
...@@ -3,7 +3,7 @@ import abc ...@@ -3,7 +3,7 @@ import abc
import yaml import yaml
import collections import collections
from functools import partial, lru_cache from functools import partial
from typing import List, Union, Dict from typing import List, Union, Dict
from lm_eval import utils from lm_eval import utils
...@@ -88,7 +88,6 @@ class TaskManager(abc.ABC): ...@@ -88,7 +88,6 @@ class TaskManager(abc.ABC):
assert self._name_is_task(name) == False assert self._name_is_task(name) == False
return self.ALL_TASKS[name]["task"] return self.ALL_TASKS[name]["task"]
@lru_cache(None)
def _load_individual_task_or_group(self, name_or_config: Union[str, dict] = None, parent_name: str = None) -> ConfigurableTask: def _load_individual_task_or_group(self, name_or_config: Union[str, dict] = None, parent_name: str = None) -> ConfigurableTask:
def load_task(config, task, group=None): def load_task(config, task, group=None):
...@@ -118,12 +117,13 @@ class TaskManager(abc.ABC): ...@@ -118,12 +117,13 @@ class TaskManager(abc.ABC):
} }
else: else:
task_config = name_or_config task_config = name_or_config
return load_task(task_config, task=name_or_config, group=parent_name) return load_task(task_config, task=task_name, group=parent_name)
else: else:
group_name = name_or_config["group"] group_name = name_or_config["group"]
subtask_list = name_or_config["task"] subtask_list = name_or_config["task"]
if self._get_yaml_path(group_name) == -1: # What if group name is not registed/is new?
if (self._name_is_registered(group_name) is False) or (self._get_yaml_path(group_name) == -1):
all_subtasks = {group_name: (parent_name, None)} all_subtasks = {group_name: (parent_name, None)}
else: else:
all_subtasks = {} all_subtasks = {}
......
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