Better error message

microsoft · Nov 29, 2021 · d854df8 · d854df8
1 parent cc80b87
commit d854df8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/torchgeo/datasets/spacenet.py b/torchgeo/datasets/spacenet.py
@@ -109,7 +109,9 @@ def __init__(
         if to_be_downloaded:
             if not download:
                 raise RuntimeError(
-                    "Dataset not found. You can use download=True to download it."
+                    f"Dataset not found in `root={self.root}` and `download=False`, "
+                    "either specify a different `root` directory or use "
+                    "`download=True` to automaticaly download the dataset."
                 )
             else:
                 self._download(to_be_downloaded, api_key)
@@ -1071,7 +1073,9 @@ def __init__(
         if to_be_downloaded:
             if not download:
                 raise RuntimeError(
-                    "Dataset not found. You can use download=True to download it."
+                    f"Dataset not found in `root={self.root}` and `download=False`, "
+                    "either specify a different `root` directory or use "
+                    "`download=True` to automaticaly download the dataset."
                 )
             else:
                 self._download(to_be_downloaded, api_key)