corrected order of valid and train data arguments in DomainDataModule (…

…#82) * swap order of valid and train arguments for class DomainDataModule * also change for GWDataModule --------- Co-authored-by: bdvllrs <[email protected]>
ruflab · Sep 19, 2024 · 48dc6ca · 48dc6ca
1 parent 7d90ee9
commit 48dc6ca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/shimmer_basics.md b/docs/shimmer_basics.md
@@ -65,8 +65,8 @@ from torch.utils.data import DataLoader, TensorDataset
 class DomainDataModule(LightningDataModule):
     def __init__(
         self,
-        val_dataset: torch.Tensor,
         train_dataset: torch.Tensor,
+        val_dataset: torch.Tensor,
         batch_size: int,
     ) -> None:
         super().__init__()
@@ -328,8 +328,8 @@ from shimmer import RepeatedDataset
 class GWDataModule(LightningDataModule):
     def __init__(
         self,
-        val_datasets: dict[frozenset[str], DomainDataset],
         train_datasets: dict[frozenset[str], DomainDataset],
+        val_datasets: dict[frozenset[str], DomainDataset],
         batch_size: int,
     ) -> None:
         super().__init__()