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 .'} ```