Allow more dynamic shape scans to be compiled

[ricardoV94]

Description

Related Issue

• Closes #
• Related to local_subtensor_merge can complicate graphs  #112

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):

[jessegrabowski]

Are you also going to fix the [right] return value from the shape_i rewrite in this PR?

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 80.83%. Comparing base (aad78d5) to head (5ee6c5d).

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #647   +/-   ##
=======================================
  Coverage   80.82%   80.83%           
=======================================
  Files         162      162           
  Lines       46805    46815   +10     
  Branches    11439    11442    +3     
=======================================
+ Hits        37832    37842   +10     
  Misses       6710     6710           
  Partials     2263     2263           

Files	Coverage Δ
pytensor/tensor/rewriting/basic.py	94.17% <100.00%> (+0.10%)	⬆️

[ricardoV94]

Are you also going to fix the [right] return value from the shape_i rewrite in this PR?

That was not an error actually, I explained in DM on Slack