Skip to content

Commit 011b8b0

Browse files
authored
🐛 fix(arrayish): unary generic mixin (#123)
1 parent ccd39b8 commit 011b8b0

File tree

1 file changed

+4
-2
lines changed
  • src/quaxed/experimental/_arrayish

1 file changed

+4
-2
lines changed

src/quaxed/experimental/_arrayish/unary.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -220,9 +220,11 @@ def __abs__(self) -> R:
220220
# Combined Mixins
221221

222222

223-
class LaxUnaryMixin(LaxPosMixin, LaxNegMixin, LaxAbsMixin):
223+
class LaxUnaryMixin(LaxPosMixin, LaxNegMixin[R], LaxAbsMixin[R]):
224224
"""Combined mixin for unary operations using quaxified `jax.lax`."""
225225

226226

227-
class NumpyUnaryMixin(NumpyPosMixin, NumpyNegMixin, NumpyAbsMixin, NumpyInvertMixin):
227+
class NumpyUnaryMixin(
228+
NumpyPosMixin, NumpyNegMixin[R], NumpyAbsMixin[R], NumpyInvertMixin[R]
229+
):
228230
"""Combined mixin for unary operations using quaxified `jax.numpy`."""

0 commit comments

Comments
 (0)