Skip to content

Commit c4e1c5c

Browse files
authored
Merge pull request #428 from PyAutoLabs/feature/jax-guard-pointer-retarget
docs: retarget xp=np guard ValueError pointer to using_jax.py
2 parents af940a7 + 05c6d67 commit c4e1c5c

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

autoarray/structures/decorators/abstract.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,8 @@ def __init__(self, func, obj, grid, xp=np, *args, **kwargs):
5454
f"Called {getattr(func, '__qualname__', repr(func))} with "
5555
f"xp=np but the input grid is JAX-backed (grid.use_jax=True). "
5656
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."
57+
f"See the autolens_workspace `scripts/guides/using_jax.py` "
58+
f"guide for the JIT-it-yourself pattern."
5959
)
6060

6161
self.func = func

0 commit comments

Comments
 (0)