Fix support for old and new style numpy exceptions

I don't understand why `np.AxisError` doesn't work for np 2.x
GeoStat-Framework · Nov 20, 2024 · 791256d · 791256d
1 parent ec9a16c
commit 791256d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/gstools/field/pgs.py b/src/gstools/field/pgs.py
@@ -70,6 +70,11 @@ def calc_pgs(self):
         pgs : :class:`numpy.ndarray`
             the plurigaussian field
         """
+        # very clunky way of supporting both np 1.x and 2.x exceptions
+        try:
+            np.AxisError = np.exceptions.AxisError
+        except:
+            ...
         try:
             mapping = np.stack(self._Zs, axis=1)
         except np.AxisError: