From 85bb9c9132b83b45b48deeb33e4070b2c34db5dc Mon Sep 17 00:00:00 2001 From: rijuld Date: Wed, 8 Jan 2025 22:50:20 +0530 Subject: [PATCH] Added parent class --- torchgeo/datamodules/substation.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torchgeo/datamodules/substation.py b/torchgeo/datamodules/substation.py index 6aa3f01498..038da7498c 100644 --- 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