From 2596f226f45e6940390d24b46120116eb2f1b526 Mon Sep 17 00:00:00 2001 From: Linchen Xiao Date: Thu, 15 Aug 2024 11:30:30 +0800 Subject: [PATCH] [Fix] longbench dataset load fix (#1422) --- opencompass/datasets/longbench/longbench_vcsum.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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 = []