From b9eeb512ff0d2f2e0ec9449e742726b7cc335dd4 Mon Sep 17 00:00:00 2001 From: Claude Date: Sun, 19 Jul 2026 15:05:57 +0000 Subject: [PATCH] fix: stop clobbering the working tfp-nightly in the smoke installer MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit The smoke installer ran `pip install tensorflow-probability==0.25.0` after `PyAutoArray[optional]`, overwriting the working `tfp-nightly` (which PyAutoArray[optional] pins) with the stable release. Stable tfp crashes at import under the resolved modern JAX — it references `jax.interpreters.xla.pytype_aval_mappings`, removed from JAX >=0.10 — which broke the JAX Matern-kernel (`delaunay_mge`) likelihood path. Remove the line and document why; the JAX path now imports the compatible tfp-nightly. Co-Authored-By: Claude Opus 4.8 Claude-Session: https://claude.ai/code/session_013ciVftxvYpefh59wSkR7jN --- .github/scripts/smoke_install.sh | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/.github/scripts/smoke_install.sh b/.github/scripts/smoke_install.sh index f37e1df..ab7693d 100755 --- a/.github/scripts/smoke_install.sh +++ b/.github/scripts/smoke_install.sh @@ -13,4 +13,8 @@ else pip install ./PyAutoNerves ./PyAutoFit ./PyAutoArray ./PyAutoGalaxy ./PyAutoLens pip install numba fi -pip install tensorflow-probability==0.25.0 +# NOTE: do NOT `pip install tensorflow-probability==0.25.0` here. The stable +# release crashes at import under the resolved modern JAX +# (`jax.interpreters.xla.pytype_aval_mappings` was removed), which broke the +# JAX Matern-kernel (delaunay_mge) likelihood path. The working modified-Bessel +# dependency is `tfp-nightly`, pinned by `PyAutoArray[optional]` above.