This commit is contained in:
Myhs-phz 2025-03-05 09:46:14 +00:00
parent 28d179c540
commit 2cd27813ef
2 changed files with 6 additions and 6 deletions

View File

@ -12,8 +12,8 @@ from mmengine.config import Config, DictAction
from opencompass.registry import PARTITIONERS, RUNNERS, build_from_cfg
from opencompass.runners import SlurmRunner
from opencompass.summarizers import DefaultSummarizer
from opencompass.utils import (LarkReporter, read_from_station,
save_to_station, get_logger)
from opencompass.utils import (LarkReporter, get_logger, read_from_station,
save_to_station)
from opencompass.utils.run import (fill_eval_cfg, fill_infer_cfg,
get_config_from_arg)

View File

@ -123,8 +123,8 @@ def save_to_station(cfg, args):
dataset_list = [[
dataset_abbr_from_cfg(dataset),
[dataset_abbr_from_cfg(base) for base in dataset['base_models']]
] if 'base_models' in dataset.keys()
else [dataset_abbr_from_cfg(dataset), ['']]
] if 'base_models' in dataset.keys() else
[dataset_abbr_from_cfg(dataset), ['']]
for dataset in cfg['datasets']]
rs_exist_results = []
@ -307,8 +307,8 @@ def read_from_station(cfg, args):
dataset_list = [[
dataset_abbr_from_cfg(dataset),
[dataset_abbr_from_cfg(base) for base in dataset['base_models']]
] if 'base_models' in dataset.keys()
else [dataset_abbr_from_cfg(dataset), ['']]
] if 'base_models' in dataset.keys() else
[dataset_abbr_from_cfg(dataset), ['']]
for dataset in cfg['datasets']]
existing_results_list = []