Update blackjax version to avoid TypeError on deprecated kwarg

[ricardoV94]

Description

Blackjax tests started failing from an issue that was solved in blackjax-devs/blackjax#664

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify):

---

📚 Documentation preview 📚: https://pymc--7317.org.readthedocs.build/en/7317/

[ricardoV94]

It seems that the newer jaxlib which blackjax requires is not available yet on conda.

Do you happen to know something about this @junpenglao (or if I missed something obvious)

[ricardoV94]

We may need to mark as xfail until we can install from conda

[twiecki]

Seems like jaxlib builds are broken currently. conda-forge/jaxlib-feedstock#249

[ricardoV94]

I don't understand why did blackjax start failing if we had everything jax/jaxlib/blackjax pinned to a specific version?

Conda-forge still seems stuck at the old v0.4.23 version of jaxlib unfortunately

[ricardoV94]

We should be printing the installed packages in the CI to see what actually got installed

[ricardoV94]

It's installing the wrong version of blackjax ...

[ricardoV94]

Ah we only lower pinned it!

[ricardoV94]

Closing in favor of #7320