From 9a78733f914f8c3223e069ab83f0b3b90547fba9 Mon Sep 17 00:00:00 2001 From: Anton Eriksson Date: Tue, 8 Oct 2024 19:19:58 +0200 Subject: [PATCH] fix DataModule docstring --- torchgeo/datamodules/geo.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torchgeo/datamodules/geo.py b/torchgeo/datamodules/geo.py index cd21d8ac546..38bf0ad4e9b 100644 --- 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__()