(ds: Dataset)
| 450 | |
| 451 | |
| 452 | def _get_dataset(ds: Dataset) -> list[TauBenchDataInst]: |
| 453 | task_ids = ds.indexes or list(range(len(_DATASET_SPLITS[ds.split]))) |
| 454 | if ds.max_size is not None: |
| 455 | task_ids = task_ids[: ds.max_size] |
| 456 | random.shuffle(task_ids) |
| 457 | return task_ids |
| 458 | |
| 459 | |
| 460 | def _get_datasets( |
no outgoing calls
no test coverage detected