Don't return tuple from jax.scipy.linalg.qr when mode = 'r'

[jessegrabowski]

Description

Fixes a bug uncovered by pymc-devs/pymc-extras#597 . When we switched from jax.numpy.linalg.qr to jax.scipy.linalg.qr, the return type changed from a matrix to a 1-tuple when mode='r'.

This PR fixes this by unpacking the return of jax.scipy.linalg.qr when only one value is returned.

Related Issue

• Closes #
• 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 (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://pytensor--1682.org.readthedocs.build/en/1682/

[ricardoV94]

I'm gonna trust you ran pre-commit since the CI one is not running :)

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 81.62%. Comparing base (792bd04) to head (a3a7c4c).
⚠️ Report is 7 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1682      +/-   ##
==========================================
- Coverage   81.64%   81.62%   -0.02%     
==========================================
  Files         242      242              
  Lines       53573    53545      -28     
  Branches     9453     9442      -11     
==========================================
- Hits        43737    43708      -29     
- Misses       7358     7361       +3     
+ Partials     2478     2476       -2     

Files with missing lines	Coverage Δ
pytensor/link/jax/dispatch/slinalg.py	95.14% <100.00%> (+0.04%)	⬆️

🚀 New features to boost your workflow:

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

[ricardoV94]

@jessegrabowski you're a naughty boy, opening PRs on the pymc-devs branch instead of your own fork

[jessegrabowski]

Ah my bad, thanks for pointing that out. I'll fix it.