From c850734aa27c57cd6b0238dd0d4627cbfcccbff1 Mon Sep 17 00:00:00 2001 From: Flaick <471570879@qq.com> Date: Fri, 9 May 2025 12:44:31 +0000 Subject: [PATCH] set up default category value for hle --- opencompass/datasets/hle.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/opencompass/datasets/hle.py b/opencompass/datasets/hle.py index 802c667d..b14507cc 100644 --- a/opencompass/datasets/hle.py +++ b/opencompass/datasets/hle.py @@ -9,7 +9,7 @@ from .base import BaseDataset class HLEDataset(BaseDataset): @staticmethod - def load(path: str, category: str): + def load(path: str, category: str | None = None): dataset = load_dataset(path) ds = dataset['test'].filter(lambda x: x['image'] == '') if category: