fix bench

This commit is contained in:
bio-mlhui 2025-05-29 04:28:00 +00:00
parent c4f219246c
commit 37be7655f5

View File

@ -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