We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ccd39b8 commit 011b8b0Copy full SHA for 011b8b0
src/quaxed/experimental/_arrayish/unary.py
@@ -220,9 +220,11 @@ def __abs__(self) -> R:
220
# Combined Mixins
221
222
223
-class LaxUnaryMixin(LaxPosMixin, LaxNegMixin, LaxAbsMixin):
+class LaxUnaryMixin(LaxPosMixin, LaxNegMixin[R], LaxAbsMixin[R]):
224
"""Combined mixin for unary operations using quaxified `jax.lax`."""
225
226
227
-class NumpyUnaryMixin(NumpyPosMixin, NumpyNegMixin, NumpyAbsMixin, NumpyInvertMixin):
+class NumpyUnaryMixin(
228
+ NumpyPosMixin, NumpyNegMixin[R], NumpyAbsMixin[R], NumpyInvertMixin[R]
229
+):
230
"""Combined mixin for unary operations using quaxified `jax.numpy`."""
0 commit comments