jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix Bug：Fix GAIA datasets lint error

Browse Source
This commit is contained in:
Deadwalk 2025-04-26 11:30:24 +08:00
parent 8c5b04947f
commit e373a2bbcf
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
opencompass/datasets/__init__.py
View File

@ -49,9 +49,9 @@ from .ds1000_interpreter import * # noqa: F401, F403
from .eprstmt import * # noqa: F401, F403
from .FinanceIQ import * # noqa: F401, F403
from .flores import * # noqa: F401, F403
from .gaia import * # noqa: F401, F403
from .game24 import * # noqa: F401, F403
from .gaokao_math import * # noqa: F401, F403
from .gaia import * # noqa: F401, F403
from .GaokaoBench import * # noqa: F401, F403
from .generic import * # noqa: F401, F403
from .govrepcrs import * # noqa: F401, F403

12
opencompass/datasets/gaia.py
View File

@ -1,11 +1,10 @@
import json
from os import environ
import os
from os import environ
from datasets import Dataset
from opencompass.registry import LOAD_DATASET
from opencompass.utils import get_data_path
from opencompass.utils.datasets_info import DATASETS_MAPPING
from .base import BaseDataset
@ -33,7 +32,7 @@ class GAIADataset(BaseDataset):
'level': item['Level']
})
except Exception as e:
print(f"Error loading local file: {e}")
print(f'Error loading local file: {e}')
else:
# 从本地读取
compass_data_cache = os.environ.get('COMPASS_DATA_CACHE')
@ -49,12 +48,13 @@ class GAIADataset(BaseDataset):
'file_name': line['file_name'],
'level': line['Level']
}
# 只有在file_name不为空时设置file_path
if line['file_name']:
row_data['file_path'] = f'{local_path}/{line["file_name"]}'
file_name = line['file_name']
row_data['file_path'] = f'{local_path}/{file_name}'
else:
row_data['file_path'] = ''
rows.append(row_data)
return Dataset.from_list(rows)