diff --git a/tests/datamodules/test_geo.py b/tests/datamodules/test_geo.py index d1924984ea7..80e71c52a43 100644 --- a/tests/datamodules/test_geo.py +++ b/tests/datamodules/test_geo.py @@ -7,11 +7,11 @@ import pytest import torch from _pytest.fixtures import SubRequest +from geopandas import GeoDataFrame from lightning.pytorch import Trainer from matplotlib.figure import Figure from rasterio.crs import CRS from torch import Tensor -from geopandas import GeoDataFrame from torchgeo.datamodules import ( GeoDataModule,