Skip to content

Commit 455707e

Browse files
Rename tag_solve_triangular to generic_solve_to_solve_triangular
1 parent 1d46aef commit 455707e

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

pytensor/tensor/rewriting/linalg.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def inv_as_solve(fgraph, node):
5050
@register_stabilize
5151
@register_canonicalize
5252
@node_rewriter([Solve])
53-
def tag_solve_triangular(fgraph, node):
53+
def generic_solve_to_solve_triangular(fgraph, node):
5454
"""
5555
If any solve() is applied to the output of a cholesky op, then
5656
replace it with a triangular solve.

tests/tensor/rewriting/test_linalg.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def test_transinv_to_invtrans():
8282
assert node.inputs[0].name == "X"
8383

8484

85-
def test_tag_solve_triangular():
85+
def test_generic_solve_to_solve_triangular():
8686
cholesky_lower = Cholesky(lower=True)
8787
cholesky_upper = Cholesky(lower=False)
8888
A = matrix("A")

0 commit comments

Comments
 (0)