🐛 fix(arrayish): unary generic mixin (#123)

GalacticDynamics · Jan 21, 2025 · 011b8b0 · 011b8b0
1 parent ccd39b8
commit 011b8b0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/quaxed/experimental/_arrayish/unary.py b/src/quaxed/experimental/_arrayish/unary.py
@@ -220,9 +220,11 @@ def __abs__(self) -> R:
 # Combined Mixins
 
 
-class LaxUnaryMixin(LaxPosMixin, LaxNegMixin, LaxAbsMixin):
+class LaxUnaryMixin(LaxPosMixin, LaxNegMixin[R], LaxAbsMixin[R]):
     """Combined mixin for unary operations using quaxified `jax.lax`."""
 
 
-class NumpyUnaryMixin(NumpyPosMixin, NumpyNegMixin, NumpyAbsMixin, NumpyInvertMixin):
+class NumpyUnaryMixin(
+    NumpyPosMixin, NumpyNegMixin[R], NumpyAbsMixin[R], NumpyInvertMixin[R]
+):
     """Combined mixin for unary operations using quaxified `jax.numpy`."""