handle scalars

pydata · Aug 21, 2024 · 2e6f3da · 2e6f3da
1 parent 5178661
commit 2e6f3da
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/xarray/namedarray/core.py b/xarray/namedarray/core.py
@@ -668,10 +668,10 @@ def __pos__(self, /):
 
         return positive(self)
 
-    def __add__(self, other, /):
-        from xarray.namedarray._array_api import add
+    def __add__(self, other: int | float | NamedArray, /) -> NamedArray:
+        from xarray.namedarray._array_api import add, asarray
 
-        return add(self, other)
+        return add(self, asarray(other))
 
     def __sub__(self, other, /):
         from xarray.namedarray._array_api import subtract
@@ -743,11 +743,10 @@ def __rshift__(self, other, /):
         return bitwise_right_shift(self)
 
     # Comparison Operators
+    def __eq__(self, other: int | float | bool | NamedArray, /) -> NamedArray:
+        from xarray.namedarray._array_api import equal, asarray
 
-    def __eq__(self, other, /):
-        from xarray.namedarray._array_api import equal
-
-        return equal(self, other)
+        return equal(self, asarray(other))
 
     def __ge__(self, other, /):
         from xarray.namedarray._array_api import greater_equal