There was an error while loading. Please reload this page.
1 parent af940a7 commit 05c6d67Copy full SHA for 05c6d67
1 file changed
autoarray/structures/decorators/abstract.py
@@ -54,8 +54,8 @@ def __init__(self, func, obj, grid, xp=np, *args, **kwargs):
54
f"Called {getattr(func, '__qualname__', repr(func))} with "
55
f"xp=np but the input grid is JAX-backed (grid.use_jax=True). "
56
f"Inside @jax.jit, pass xp=jnp explicitly to the library call. "
57
- f"See the autolens_workspace `scripts/guides/lens_calc.py` "
58
- f"`__JAX__` section (Phase 5d) for the JIT-it-yourself pattern."
+ f"See the autolens_workspace `scripts/guides/using_jax.py` "
+ f"guide for the JIT-it-yourself pattern."
59
)
60
61
self.func = func
0 commit comments