Make try-catch more targeted

microsoft · Jun 23, 2023 · 7ca3cb7 · 7ca3cb7
1 parent f0d9475
commit 7ca3cb7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/torchgeo/datasets/utils.py b/torchgeo/datasets/utils.py
@@ -762,8 +762,9 @@ def _listdir_vsi_recursive(root: str) -> list[str]:
         try:
             subdirs = fiona.listdir(dir)
             dirs.extend([os.path.join(dir, subdir) for subdir in subdirs])
-        except FionaValueError:
-            files.append(dir)
+        except FionaValueError as e:
+            if "is not a directory" in str(e):
+                files.append(dir)
     return files