🐛 fix(arrayish): enable add -> radd (#128)

GalacticDynamics · Feb 7, 2025 · 973ddd7 · 973ddd7
1 parent 1d15e61
commit 973ddd7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/src/quaxed/experimental/_arrayish/binary.py b/src/quaxed/experimental/_arrayish/binary.py
@@ -104,7 +104,10 @@ class LaxAddMixin(Generic[T, R]):
     """
 
     def __add__(self, other: T) -> R:
-        return qlax.add(self, other)
+        try:
+            return qlax.add(self, other)
+        except Exception:  # noqa: BLE001
+            return NotImplemented
 
 
 class NumpyAddMixin(Generic[T, R]):
@@ -126,7 +129,10 @@ class NumpyAddMixin(Generic[T, R]):
     """
 
     def __add__(self, other: T) -> R:
-        return qnp.add(self, other)
+        try:
+            return qnp.add(self, other)
+        except Exception:  # noqa: BLE001
+            return NotImplemented
 
 
 # -------------------------------------
@@ -152,7 +158,10 @@ class LaxRAddMixin(Generic[T, R]):
     """
 
     def __radd__(self, other: T) -> R:
-        return qlax.add(other, self)
+        try:
+            return qlax.add(other, self)
+        except Exception:  # noqa: BLE001
+            return NotImplemented
 
 
 class NumpyRAddMixin(Generic[T, R]):
@@ -174,7 +183,10 @@ class NumpyRAddMixin(Generic[T, R]):
     """
 
     def __radd__(self, other: T) -> R:
-        return qnp.add(other, self)
+        try:
+            return qnp.add(other, self)
+        except Exception:  # noqa: BLE001
+            return NotImplemented
 
 
 # -------------------------------------

diff --git a/uv.lock b/uv.lock