Add rewrites for inv(diag(x)) and inv(eye)

[tanish1729]

Description

Added rewrites for inv(diag(x)) = 1/x and rewrites inv(eye) -> eye

Related Issue

• Related to ENH: Implement COLA library rewrites for linear algebra functions #573

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

Attention: Patch coverage is 94.59459% with 2 lines in your changes missing coverage. Please review.

Project coverage is 81.70%. Comparing base (b65d08c) to head (370f2d8).
Report is 132 commits behind head on main.

Files with missing lines	Patch %	Lines
pytensor/tensor/rewriting/linalg.py	94.59%	1 Missing and 1 partial ⚠️

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #898   +/-   ##
=======================================
  Coverage   81.69%   81.70%           
=======================================
  Files         182      182           
  Lines       47585    47626   +41     
  Branches    11584    11601   +17     
=======================================
+ Hits        38875    38913   +38     
- Misses       6520     6521    +1     
- Partials     2190     2192    +2     

Files with missing lines	Coverage Δ
pytensor/tensor/rewriting/linalg.py	91.66% <94.59%> (+0.38%)	⬆️

... and 4 files with indirect coverage changes

[jessegrabowski]

Getting close! Can you also add a test of cases that should not be rewritten, for example eye with k!=0. Check the different off-ramps in the rewrites and make sure they happen.

[jessegrabowski]

Import what you need directly from the various packages within pytensor.tensor, it makes the code easier to follow for people in the future

[tanish1729]

ah that makes sense. i'll do that from next time cuz the test file has way too many uses of pt rn 😓

[jessegrabowski]

Small change, then I think this is ready