mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
fix bench
This commit is contained in:
parent
c4f219246c
commit
37be7655f5
@ -247,7 +247,7 @@ class HealthBenchDataset(BaseDataset):
|
||||
case _:
|
||||
raise Exception(f'Unrecognized subset type: {subset}') # noqa: W291, E501
|
||||
dataset = load_dataset(path, data_files=data_files, split='test')
|
||||
dataset = dataset.select(range(2))
|
||||
# dataset = dataset.select(range(2))
|
||||
dataset = dataset.map(lambda item: _parse(item))
|
||||
|
||||
return dataset
|
||||
|
Loading…
Reference in New Issue
Block a user