improve docstring

ArneBinder · Sep 30, 2024 · e155af2 · e155af2
1 parent 4dc1270
commit e155af2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/pie_datasets/core/dataset_dict.py b/src/pie_datasets/core/dataset_dict.py
@@ -372,7 +372,8 @@ def map(  # type: ignore
                 function signature. If this is not possible, the document type of the input dataset
                 is used.
             set_batch_size_to_split_size: If enabled, set the batch_size to the size of the respective split
-                when calling map() on it. This is useful to transform whole splits.
+                when calling map() on it. This is useful to transform whole splits when using it in
+                combination with batched=True.
             **kwargs: additional keyword arguments for `datasets.Dataset.map()`
         """