Switch to relative imports

microsoft · Jan 6, 2025 · 47fcabe · 47fcabe
1 parent bb57173
commit 47fcabe
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 9 deletions.
diff --git a/torchgeo/datasets/idtrees.py b/torchgeo/datasets/idtrees.py
@@ -529,7 +529,9 @@ def normalize(x: Tensor) -> Tensor:
                 else None
             )
             preds = draw_bounding_boxes(
-                image=sample['image'], boxes=sample['prediction_bbox_xyxy'], labels=labels
+                image=sample['image'],
+                boxes=sample['prediction_bbox_xyxy'],
+                labels=labels,
             )
             preds = preds.permute((1, 2, 0)).numpy()
 

diff --git a/torchgeo/datasets/zuericrop.py b/torchgeo/datasets/zuericrop.py
@@ -11,14 +11,9 @@
 from matplotlib.figure import Figure
 from torch import Tensor
 
-from torchgeo.datasets.errors import DatasetNotFoundError, RGBBandsMissingError
-from torchgeo.datasets.geo import NonGeoDataset
-from torchgeo.datasets.utils import (
-    Path,
-    download_url,
-    lazy_import,
-    percentile_normalization,
-)
+from .errors import DatasetNotFoundError, RGBBandsMissingError
+from .geo import NonGeoDataset
+from .utils import Path, download_url, lazy_import, percentile_normalization
 
 
 class ZueriCrop(NonGeoDataset):