Provide static output shape for constant arange

[Abhinav-Khot]

Description

Modified arange such that in scenarios where we are able to infer the shape from the arguments i.e the arguments are constant, type.shape is updated accordingly.

For example :

import pytensor
import pytensor.tensor as pt

print(pt.arange(1,10,1).type.shape) #Returns (9, ). Earlier used to return (None, )

Tests are also added to test the new addition.

Related Issue

• Closes pt.arange(int) has .type.shape set to None #1301
• 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--1302.org.readthedocs.build/en/1302/

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 82.00%. Comparing base (2a7f3e1) to head (11a915b).
Report is 40 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1302      +/-   ##
==========================================
+ Coverage   81.99%   82.00%   +0.01%     
==========================================
  Files         188      188              
  Lines       48553    48483      -70     
  Branches     8673     8666       -7     
==========================================
- Hits        39812    39760      -52     
+ Misses       6579     6575       -4     
+ Partials     2162     2148      -14     

Files with missing lines	Coverage Δ
pytensor/tensor/basic.py	91.16% <100.00%> (-0.20%)	⬇️

... and 27 files with indirect coverage changes

🚀 New features to boost your workflow:

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

[ricardoV94]

NIT: put this in an else branch

[Abhinav-Khot]

I guess it would become a bit more readable that way, will make this change

[ricardoV94]

Looks good, just a nitpick, that we don't need to address if you don't agree

[ricardoV94]

Thanks @Abhinav-Khot