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 _:
|
case _:
|
||||||
raise Exception(f'Unrecognized subset type: {subset}') # noqa: W291, E501
|
raise Exception(f'Unrecognized subset type: {subset}') # noqa: W291, E501
|
||||||
dataset = load_dataset(path, data_files=data_files, split='test')
|
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))
|
dataset = dataset.map(lambda item: _parse(item))
|
||||||
|
|
||||||
return dataset
|
return dataset
|
||||||
|
Loading…
Reference in New Issue
Block a user