GeoDataset: better error message when no data found (#1193)

microsoft · Apr 10, 2023 · 4dfbcd5 · 4dfbcd5
1 parent 4cb49f8
commit 4dfbcd5
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/torchgeo/datasets/geo.py b/torchgeo/datasets/geo.py
@@ -364,9 +364,10 @@ def __init__(
                     i += 1
 
         if i == 0:
-            raise FileNotFoundError(
-                f"No {self.__class__.__name__} data was found in '{root}'"
-            )
+            msg = f"No {self.__class__.__name__} data was found in `root='{self.root}'`"
+            if self.bands:
+                msg += f" with `bands={self.bands}`"
+            raise FileNotFoundError(msg)
 
         if not self.separate_files:
             self.band_indexes = None
@@ -582,9 +583,8 @@ def __init__(
                 i += 1
 
         if i == 0:
-            raise FileNotFoundError(
-                f"No {self.__class__.__name__} data was found in '{root}'"
-            )
+            msg = f"No {self.__class__.__name__} data was found in `root='{root}'`"
+            raise FileNotFoundError(msg)
 
         self._crs = crs