We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e15260d commit 456905cCopy full SHA for 456905c
autoarray/structures/decorators/abstract.py
@@ -61,7 +61,6 @@ def __init__(self, func, obj, grid, xp=np, *args, **kwargs):
61
def _xp(self):
62
if self.use_jax:
63
import jax.numpy as jnp
64
-
65
return jnp
66
return np
67
0 commit comments