diff --git a/opencompass/datasets/longbench/longbench_vcsum.py b/opencompass/datasets/longbench/longbench_vcsum.py index 2aef8fe9..28526aaa 100644 --- a/opencompass/datasets/longbench/longbench_vcsum.py +++ b/opencompass/datasets/longbench/longbench_vcsum.py @@ -11,8 +11,9 @@ class LongBenchvcsumDataset(BaseDataset): @staticmethod def load(**kwargs): - kwargs['data_files'] = get_data_path(kwargs['data_files'], - local_mode=True) + if 'data_files' in kwargs: + kwargs['data_files'] = get_data_path(kwargs['data_files'], + local_mode=True) dataset = load_dataset(**kwargs) split = 'test' raw_data = []