fix: stop clobbering working tfp-nightly in smoke installer (JAX Matern/delaunay_mge)

[Jammy2211]

Problem

The smoke CI failed on the two JAX Matern-kernel scripts:

=== Smoke test summary: 32/34 passed ===
  FAIL  jax_likelihood_functions/interferometer/delaunay_mge.py  (exit 1)
  FAIL  jax_likelihood_functions/multi/delaunay_mge.py  (exit 1)

with:

File ".../tensorflow_probability/python/internal/backend/jax/ops.py", line 681
    jax.interpreters.xla.pytype_aval_mappings[onp.ndarray])
AttributeError: module 'jax.interpreters.xla' has no attribute 'pytype_aval_mappings'

Root cause

.github/scripts/smoke_install.sh ran pip install tensorflow-probability==0.25.0 after PyAutoArray[optional]. The [optional] extra already installs the working, pinned tfp-nightly==0.26.0.dev20260713 (used by the JAX Matern kernel's bessel_kve). The trailing line then overwrote it with the stable release, which crashes at import under the resolved modern JAX (jax.interpreters.xla.pytype_aval_mappings was removed in JAX ≥ 0.10) — exactly what PyAutoArray/pyproject.toml documents.

Verified locally: jax 0.10.2 + tfp-nightly 0.26.0.dev20260713 imports cleanly and bessel_kve matches scipy.special.kve to ~2e-15 (jit + grad OK); installing tensorflow-probability==0.25.0 on top reproduces the exact traceback.

Fix

Remove the clobbering line and document why, so the JAX path keeps the compatible tfp-nightly provided by PyAutoArray[optional].

🤖 Generated with Claude Code

https://claude.ai/code/session_013ciVftxvYpefh59wSkR7jN

---

Generated by Claude Code