Filter the dataset to only select the fields specified.
(fields: Sequence[str])
| 580 | |
| 581 | |
| 582 | def select_fields(fields: Sequence[str]) -> DatasetToDatasetFn: |
| 583 | """Filter the dataset to only select the fields specified.""" |
| 584 | |
| 585 | return rekey({k: k for k in fields}, retain_original_inputs=False) |
| 586 | |
| 587 | |
| 588 | def remove_fields(fields: Sequence[str]) -> DatasetToDatasetFn: |