fix DataModule docstring

microsoft · Oct 8, 2024 · 88cedbc · 88cedbc
1 parent b5291a6
commit 88cedbc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torchgeo/datamodules/geo.py b/torchgeo/datamodules/geo.py
@@ -46,7 +46,8 @@ def __init__(
             dataset_class: Class used to instantiate a new dataset.
             batch_size: Size of each mini-batch.
             num_workers: Number of workers for parallel data loading.
-            **kwargs: Additional keyword arguments passed to ``dataset_class``
+            **kwargs: Additional keyword arguments passed to the ``DataLoader``
+                if prefixed with 'dataloader_', else passed to ``dataset_class``.
         """
         super().__init__()