Skip to content

Commit 05c6d67

Browse files
Jammy2211Jammy2211
authored andcommitted
docs: retarget xp=np guard ValueError pointer to using_jax.py (#427)
1 parent af940a7 commit 05c6d67

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)