feat: catch pyogrio errors when reading zones for geopandas>1

martibosch · Oct 22, 2024 · 913fb06 · 913fb06
1 parent 4c3955a
commit 913fb06
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pylandstats/zonal.py b/pylandstats/zonal.py
@@ -13,6 +13,11 @@
     from fiona import errors as fiona_errors
 except ImportError:
     fiona_errors = None
+try:
+    from pyogrio import errors as pyogrio_errors
+except ImportError:
+    pyogrio_errors = None
+
 
 from . import multilandscape
 from .landscape import Landscape
@@ -22,6 +27,8 @@
 ZONES_READ_ERRORS = [AttributeError, TypeError]
 if fiona_errors is not None:
     ZONES_READ_ERRORS.append(fiona_errors.DriverError)
+if pyogrio_errors is not None:
+    ZONES_READ_ERRORS.append(pyogrio_errors.DataSourceError)
 ZONES_READ_ERRORS = tuple(ZONES_READ_ERRORS)
 
 _compute_zonal_statistics_gdf_doc = """