also change for GWDataModule

ruflab · Sep 19, 2024 · 77d96cc · 77d96cc
1 parent 6dc3af6
commit 77d96cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/shimmer_basics.md b/docs/shimmer_basics.md
@@ -334,8 +334,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__()