Support xtensor inputs in check_parameters

[ishitta-iyer]

Description

Extend pymc.distributions.dist_math.check_parameters to fully support XTensorVariable expressions and conditions.
XTensor conditions are lowered before reduction with pt.all, avoiding PyTensor’s forbidden implicit XTensor-to-tensor conversion. XTensor expressions are lowered before applying CheckParameterValue, then converted back with their original dims. This also allows PyMC’s CheckParameterValue-to--inf compilation rewrite to operate correctly.
Ordinary tensor behavior remains unchanged, including mixed tensor/XTensor conditions. Focused regression tests cover valid and invalid conditions, compiled -inf replacement, return types, preserved dims, dtype and shape, mixed conditions, and Python boolean conditions.

Related Issue

• Closes check_parameters should support xtensor inputs #8398
• Related to #

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 (not applicable; this does not add or change a public API)
• 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):

[read-the-docs-community]

Documentation build overview

📚 pymc | 🛠️ Build #34075897 | 📁 Comparing 166a7e0 against latest (47bdf54)

  🔍 Preview build  

1 file changed

± glossary.html