Fix too restrictive type assert

[ricardoV94]

Introduced in f9dfe70

Causing PyMC CI to fail in pymc-devs/pymc#7400

https://github.com/pymc-devs/pymc/actions/runs/9790920723/job/27033597109?pr=7400#step:7:3094

Description

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

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 81.03%. Comparing base (df769f6) to head (6ba70b8).
Report is 117 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #880   +/-   ##
=======================================
  Coverage   81.03%   81.03%           
=======================================
  Files         169      169           
  Lines       46845    46845           
  Branches    11485    11485           
=======================================
  Hits        37959    37959           
+ Misses       6679     6677    -2     
- Partials     2207     2209    +2     

Files with missing lines	Coverage Δ
pytensor/tensor/subtensor.py	89.20% <100.00%> (ø)

... and 2 files with indirect coverage changes