np.AxisError was added in numpy 1.13

pydata · Oct 2, 2018 · ac2c568 · ac2c568
1 parent f9c4169
commit ac2c568
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/xarray/core/dask_array_compat.py b/xarray/core/dask_array_compat.py
@@ -44,7 +44,13 @@ def isin(element, test_elements, assume_unique=False, invert=False):
     import math
     from numbers import Integral, Real
 
-    AxisError = np.AxisError
+    try:
+        AxisError = np.AxisError
+    except AttributeError:
+        try:
+            np.array([0]).sum(axis=5)
+        except Exception as e:
+            AxisError = type(e)
 
     def validate_axis(axis, ndim):
         """ Validate an input to axis= keywords """