Update dataset.py corrected order of valid and train arguments in Dom…

…ainDataModule (#83) * Update dataset.py swapped valid and train arguments in datadomain see last pull request * fix also for GWDataModule --------- Co-authored-by: bdvllrs <[email protected]>
ruflab · Sep 19, 2024 · 854ea4b · 854ea4b
1 parent 48dc6ca
commit 854ea4b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/main_example/dataset.py b/examples/main_example/dataset.py
@@ -33,8 +33,8 @@ def get_domain_data(
 class DomainDataModule(LightningDataModule):
     def __init__(
         self,
-        val_dataset: torch.Tensor,
         train_dataset: torch.Tensor,
+        val_dataset: torch.Tensor,
         batch_size: int,
     ) -> None:
         super().__init__()
@@ -109,8 +109,8 @@ def make_datasets(
 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__()