Skip to content

Commit 766460a

Browse files
committed
minor changes
1 parent 47942c7 commit 766460a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pytensor/tensor/rewriting/linalg.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -583,10 +583,10 @@ def rewrite_inv_inv(fgraph, node):
583583
if potential_inner_inv is None or potential_inner_inv.op is None:
584584
return None
585585
# Check if its an inner solve as well, does that have b = eye
586-
solve_inv_check = True if hasattr(potential_inner_inv.op, "core_op") else False
587-
# if potential_inner_inv.op and isinstance(potential_inner_inv.op, DimShuffle):
588-
# return None
589-
if solve_inv_check and isinstance(potential_inner_inv.op.core_op, valid_solves):
586+
solve_inv_check = False
587+
if hasattr(potential_inner_inv.op, "core_op") and isinstance(
588+
potential_inner_inv.op.core_op, valid_solves
589+
):
590590
solve_inv_check = _find_solve_with_eye(potential_inner_inv)
591591
if not solve_inv_check:
592592
return None

0 commit comments

Comments
 (0)