improve docstring

ArneBinder · Sep 17, 2024 · 5859d92 · 5859d92
1 parent 9f888e6
commit 5859d92
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
@@ -361,7 +361,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()`
         """