From 47fcabe2563c0467be9c8f53e8e41fae375124d2 Mon Sep 17 00:00:00 2001 From: Ashwin Nair Date: Mon, 6 Jan 2025 15:58:40 +0400 Subject: [PATCH] Switch to relative imports --- torchgeo/datasets/idtrees.py | 4 +++- torchgeo/datasets/zuericrop.py | 11 +++-------- 2 files changed, 6 insertions(+), 9 deletions(-) diff --git a/torchgeo/datasets/idtrees.py b/torchgeo/datasets/idtrees.py index 0b2f0283805..e773a02d20c 100644 --- 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 index b7fd8dcff67..c8d1b103f38 100644 --- 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):