From f8a434dd5f0319c69c7c45122c5fd1b3d0f77bdf Mon Sep 17 00:00:00 2001 From: Albert Villanova del Moral <8515462+albertvillanova@users.noreply.github.com> Date: Mon, 29 Jun 2026 08:31:18 +0200 Subject: [PATCH] Fix docstring example for select_columns in IterableDatasetDict --- src/datasets/dataset_dict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/datasets/dataset_dict.py b/src/datasets/dataset_dict.py index 4abea0a381a..2ef8c7c0751 100644 --- a/src/datasets/dataset_dict.py +++ b/src/datasets/dataset_dict.py @@ -2253,7 +2253,7 @@ def select_columns(self, column_names: Union[str, list[str]]) -> "IterableDatase ```py >>> from datasets import load_dataset >>> ds = load_dataset("cornell-movie-review-data/rotten_tomatoes", streaming=True) - >>> ds = ds.select("text") + >>> ds = ds.select_columns("text") >>> next(iter(ds["train"])) {'text': 'the rock is destined to be the 21st century\'s new " conan " and that he\'s going to make a splash even greater than arnold schwarzenegger , jean-claud van damme or steven segal .'} ```