fix in datasets module to match import structure

m2lines · Nov 9, 2023 · ce70a34 · ce70a34
1 parent d3947e5
commit ce70a34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gz21_ocean_momentum/data/datasets.py b/src/gz21_ocean_momentum/data/datasets.py
@@ -188,7 +188,7 @@ def add_targets_transform(self, transform):
             )
         self.transforms["targets"].add_transform(transform)
 
-    def fit(self, x: torch.utils.data.Dataset):
+    def fit(self, x: torch.Dataset):
         """
         Call the fit method of all array transforms in the list
         of features and target transforms on the passed Dataset.