Adding Dirichlet and SimplexTransform to pymc.dims

[zaxtax]

Description

Adds support for the Dirichlet distributions in pymc.dims

Related Issue

This depends on pymc-devs/pytensor#1629 to get merged

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--7921.org.readthedocs.build/en/7921/

[ricardoV94]

If you rebase from main it shuold pick the newer PyTensor with the xtensor function you need

[codecov]

Codecov Report

❌ Patch coverage is 97.22222% with 1 line in your changes missing coverage. Please review.
✅ Project coverage is 91.47%. Comparing base (3c42a18) to head (bb3e008).
⚠️ Report is 2 commits behind head on main.

Files with missing lines	Patch %	Lines
pymc/dims/distributions/vector.py	92.85%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #7921      +/-   ##
==========================================
+ Coverage   89.33%   91.47%   +2.14%     
==========================================
  Files         116      116              
  Lines       18904    18939      +35     
==========================================
+ Hits        16887    17325     +438     
+ Misses       2017     1614     -403     

Files with missing lines	Coverage Δ
pymc/dims/distributions/transforms.py	100.00% <100.00%> (ø)
pymc/dims/distributions/vector.py	90.00% <92.85%> (+0.44%)	⬆️

... and 13 files with indirect coverage changes

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[zaxtax]

@ricardoV94 the failing test seems unrelated to my PR

[ricardoV94]

Yes it is not, it's a warning that's no longer emitted in the latest PyTensor. Can you remove it in a separate commit?