From a058b9b4931e7edb14319d2f0dc5a0627d0bf9e3 Mon Sep 17 00:00:00 2001 From: Myhs-phz Date: Wed, 5 Mar 2025 08:51:44 +0000 Subject: [PATCH] fix --- opencompass/utils/result_station.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/opencompass/utils/result_station.py b/opencompass/utils/result_station.py index 96f0eebf..1664ac3d 100644 --- a/opencompass/utils/result_station.py +++ b/opencompass/utils/result_station.py @@ -122,8 +122,8 @@ def Save_To_Station(cfg, args): dataset_list = [[ dataset_abbr_from_cfg(dataset), - [dataset_abbr_from_cfg(base) for base in dataset['given_pred']] - ] if 'base_models' in dataset.keys() and 'eval_cfg' in dataset.keys() + [dataset_abbr_from_cfg(base) for base in dataset['base_models']] + ] if 'base_models' in dataset.keys() else [dataset_abbr_from_cfg(dataset), ['']] for dataset in cfg['datasets']] @@ -306,8 +306,8 @@ def Read_From_Station(cfg, args): dataset_list = [[ dataset_abbr_from_cfg(dataset), - [dataset_abbr_from_cfg(base) for base in dataset['given_pred']] - ] if 'base_models' in dataset.keys() and 'eval_cfg' in dataset.keys() + [dataset_abbr_from_cfg(base) for base in dataset['base_models']] + ] if 'base_models' in dataset.keys() else [dataset_abbr_from_cfg(dataset), ['']] for dataset in cfg['datasets']]