resolved bugs

microsoft · Oct 21, 2024 · 10637af · 10637af
1 parent 7dff61c
commit 10637af
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/tests/datasets/test_substation_seg.py b/tests/datasets/test_substation_seg.py
@@ -6,8 +6,7 @@
 from pathlib import Path
 from pytest import MonkeyPatch
 
-from .datasets import SubstationDataset
-from .errors import DatasetNotFoundError
+from torchgeo.datasets import SubstationDataset, DatasetNotFoundError
 
 
 class TestSubstationDataset:

diff --git a/torchgeo/datasets/__init__.py b/torchgeo/datasets/__init__.py
@@ -14,7 +14,6 @@
 from .cabuar import CaBuAr
 from .cbf import CanadianBuildingFootprints
 from .cdl import CDL
-from .chabud import ChaBuDx
 from .chesapeake import (
     Chesapeake,
     ChesapeakeCVPR,