We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent dba3d18 commit 2c93dcfCopy full SHA for 2c93dcf
pytensor/tensor/rewriting/linalg.py
@@ -551,12 +551,12 @@ def _find_solve_with_eye(node):
551
return False
552
# Check whether second input to solve is Eye
553
solve_inputs = node.inputs
554
- potential_eye_input = solve_inputs[1].owner
+ eye_input = solve_inputs[1].owner
555
556
- if not (potential_eye_input and isinstance(potential_eye_input.op, Eye)):
+ if not (eye_input and isinstance(eye_input.op, Eye)):
557
558
559
- if getattr(potential_eye_input.inputs[-1], "data", -1).item() != 0:
+ if eye_input.inputs[-1].data.item() != 0:
560
561
return True
562
0 commit comments