Added parent class

microsoft · Jan 8, 2025 · 85bb9c9 · 85bb9c9
1 parent 743113e
commit 85bb9c9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torchgeo/datamodules/substation.py b/torchgeo/datamodules/substation.py
@@ -6,9 +6,10 @@
 from tqdm import tqdm
 
 from ..datasets import Substation
+from .geo import NonGeoDataModule
 
 
-class SubstationDataModule:
+class SubstationDataModule(NonGeoDataModule):
     """Substation Data Module with train-test split and transformations.
 
     .. versionadded:: 0.7