diff --git a/src/datasets/iterable_dataset.py b/src/datasets/iterable_dataset.py index 9ac87783677..a3b09561218 100644 --- a/src/datasets/iterable_dataset.py +++ b/src/datasets/iterable_dataset.py @@ -194,7 +194,10 @@ def __iter__(self): def shuffle_data_sources(self, seed: Optional[int]) -> "MappedExamplesIterable": """Shuffle the wrapped examples iterable.""" return MappedExamplesIterable( - self.ex_iterable.shuffle_data_sources(seed), function=self.function, batch_size=self.batch_size + self.ex_iterable.shuffle_data_sources(seed), + function=self.function, + batched=self.batched, + batch_size=self.batch_size, ) @property