Use old numpy exceptions

GeoStat-Framework · Nov 20, 2024 · ec9a16c · ec9a16c
1 parent 727f86d
commit ec9a16c
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/gstools/field/pgs.py b/src/gstools/field/pgs.py
@@ -70,7 +70,16 @@ def calc_pgs(self):
         pgs : :class:`numpy.ndarray`
             the plurigaussian field
         """
-        mapping = np.stack(self._Zs, axis=1)
+        try:
+            mapping = np.stack(self._Zs, axis=1)
+        except np.AxisError:
+            # if dim==1, `fields` is prob. a raw field & not a 1-tuple or
+            # equivalent
+            if self._dim == 1:
+                self._Zs = [self._Zs]
+                mapping = np.stack(self._Zs, axis=1)
+            else:
+                raise
         pos_l = []
         for d in range(self._dim):
             pos_l.append(