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

set up default category value for hle

Browse Source
This commit is contained in:
Flaick 2025-05-09 12:44:31 +00:00
parent ef3ca3ebc1
commit c850734aa2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
opencompass/datasets/hle.py
View File

@ -9,7 +9,7 @@ from .base import BaseDataset
class HLEDataset(BaseDataset): class HLEDataset(BaseDataset):
@staticmethod @staticmethod
def load(path: str, category: str): def load(path: str, category: str | None = None):
dataset = load_dataset(path) dataset = load_dataset(path)
ds = dataset['test'].filter(lambda x: x['image'] == '') ds = dataset['test'].filter(lambda x: x['image'] == '')
if category: if category: