Commit 538ccab
feat: SimulatorImaging.via_tracer_from auto-defaults xp from parent use_jax
Changes via_tracer_from(xp=np) to via_tracer_from(xp=None) with auto-fallback
to self._xp inherited from aa.SimulatorImaging. When the simulator is
constructed with use_jax=True, via_tracer_from now routes xp=jnp through
tracer.padded_image_2d_from and via_image_from without the user passing it
explicitly.
Depends on Jammy2211/PyAutoArray PR adding the use_jax constructor flag +
_xp property to the parent aa.SimulatorImaging.
Part of Phase 2 PR 2 of z_features/jax_user_intro.md.
Issue: PyAutoArray#334
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>1 parent a90f888 commit 538ccab
1 file changed
Lines changed: 4 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
22 | 22 | | |
23 | 23 | | |
24 | 24 | | |
25 | | - | |
| 25 | + | |
26 | 26 | | |
27 | 27 | | |
28 | 28 | | |
| |||
50 | 50 | | |
51 | 51 | | |
52 | 52 | | |
| 53 | + | |
| 54 | + | |
| 55 | + | |
53 | 56 | | |
54 | 57 | | |
55 | 58 | | |
| |||
0 commit comments