Skip to content

Pin tfp-nightly to prevent recurrence of #428 and fix #443#444

Open
retconrecon wants to merge 2 commits intoprobml:mainfrom
retconrecon:fix/tfp-nightly-jax07-compat
Open

Pin tfp-nightly to prevent recurrence of #428 and fix #443#444
retconrecon wants to merge 2 commits intoprobml:mainfrom
retconrecon:fix/tfp-nightly-jax07-compat

Conversation

@retconrecon
Copy link

Fixes #428
Fixes #443

Pins tfp-nightly to ==0.26.0.dev20260101 instead of leaving it unpinned.

Why: Nightly builds aren't QA-gated, so an unpinned tfp-nightly could resolve to a different build on any given day — the exposure window to breaking changes is daily rather than per-stable-release. Pinning to a verified build preserves reproducibility and reduces the risk of a similar breakage recurring from an upstream change.

Verified against:

  • JAX 0.8.1
  • tfp-nightly 0.26.0.dev20260101
  • NumPy 2.3.5
  • All dynamax imports, ARHMM/LGSSM initialization, Kalman filtering, and GaussianHMM initialization pass.

When TFP ships a stable release with JAX 0.7+ support, this can be reverted back to tensorflow_probability.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

GaussianHMM: TypeError: reshape() got an unexpected keyword argument 'newshape' Tests failing due to incompatibility between TFP and JAX 0.7

1 participant