Add array_namespace

andersy005 · Sep 15, 2023 · cf91823 · cf91823
1 parent fb2ca4d
commit cf91823
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/xarray/tests/test_dataset.py b/xarray/tests/test_dataset.py
@@ -415,6 +415,9 @@ def __init__(self):
             def __array_function__(self, *args, **kwargs):
                 pass
 
+            def __array_namespace__(self, *args, **kwargs):
+                pass
+
             def __repr__(self):
                 return "Custom\nArray"
 

diff --git a/xarray/tests/test_formatting.py b/xarray/tests/test_formatting.py
@@ -549,10 +549,7 @@ def _repr_inline_(self, width):
 
             return formatted
 
-        def __array_function__(self, *args, **kwargs):
-            return NotImplemented
-
-        def __array_ufunc__(self, *args, **kwargs):
+        def __array_namespace__(self, *args, **kwargs):
             return NotImplemented
 
         @property