Handle numpy missing the array api function astype (#8315)

pydata · Oct 16, 2023 · 4520ce9 · 4520ce9
1 parent dafd726
commit 4520ce9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/xarray/core/duck_array_ops.py b/xarray/core/duck_array_ops.py
@@ -184,6 +184,9 @@ def cumulative_trapezoid(y, x, axis):
 def astype(data, dtype, **kwargs):
     if hasattr(data, "__array_namespace__"):
         xp = get_array_namespace(data)
+        if xp == np:
+            # numpy currently doesn't have a astype:
+            return data.astype(dtype, **kwargs)
         return xp.astype(data, dtype, **kwargs)
     return data.astype(dtype, **kwargs)