https://huggingface.co/docs/datasets/package_reference/main_classes.html#datasets.Dataset.map