rasdani/verifiable-coding-problems-python_decontaminated_fewer_test_cases

import datasets import random def limit_test_cases_uniformly(example, max_test_cases=6): num_test_cases = random.randint(1, max_test_cases) example['verification_info']['test_cases'] = example['verification_info']['test_cases'][:num_test_cases] return example ds = datasets.load_dataset("open-r1/verifiable-coding-problems-python_decontaminated", split="train") ds_filtered = ds.map(limit_test_cases_uniformly, num_proc=10)… See the full description on the dataset page: https://huggingface.co/datasets/rasdani/verifiable-coding-problems-python_decontaminated_fewer_test_cases.