Add test for `freeze_rv_and_dims` in JAX backend - Githubissues

pymc-devs / pymc

Bayesian Modeling and Probabilistic Programming in Python

https://docs.pymc.io/

Other

8.47k stars 1.97k forks source link

Add test for `freeze_rv_and_dims` in JAX backend #7324

Closed ricardoV94 closed 1 month ago

ricardoV94 commented 1 month ago

Description

This test requires https://github.com/pymc-devs/pytensor/pull/774 to be included in the pinned version of PyTensor

Checklist

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

Type of change

[ ] New feature / enhancement
[ ] Bug fix
[ ] Documentation
[x] Maintenance
[ ] Other (please specify):

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