diff --git a/opencompass/cli/main.py b/opencompass/cli/main.py index b50e88e5..d3fbc98a 100644 --- a/opencompass/cli/main.py +++ b/opencompass/cli/main.py @@ -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) diff --git a/opencompass/utils/result_station.py b/opencompass/utils/result_station.py index 3fc7f471..7ad6b9eb 100644 --- a/opencompass/utils/result_station.py +++ b/opencompass/utils/result_station.py @@ -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 = []