Skip to content

Commit 1d46aef

Browse files
run pre-commit hooks
1 parent 5ab280c commit 1d46aef

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

tests/tensor/rewriting/test_linalg.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,9 @@ def test_tag_solve_triangular():
106106
assert not any(isinstance(op, Solve) for op in op_list)
107107
assert any(isinstance(op, SolveTriangular) for op in op_list)
108108

109-
assert_allclose(f(X, eye) @ X_chol, eye, atol=1e-8 if config.floatX.endswith('64') else 1e-4)
109+
assert_allclose(
110+
f(X, eye) @ X_chol, eye, atol=1e-8 if config.floatX.endswith("64") else 1e-4
111+
)
110112

111113
f = pytensor.function([A, x], b2)
112114

@@ -115,7 +117,11 @@ def test_tag_solve_triangular():
115117
op_list = [node.op for node in toposort]
116118
assert not any(isinstance(op, Solve) for op in op_list)
117119
assert any(isinstance(op, SolveTriangular) for op in op_list)
118-
assert_allclose(f(X, eye).T @ X_chol, eye, atol=1e-8 if config.floatX.endswith('64') else 1e-4)
120+
assert_allclose(
121+
f(X, eye).T @ X_chol,
122+
eye,
123+
atol=1e-8 if config.floatX.endswith("64") else 1e-4,
124+
)
119125

120126

121127
def test_matrix_inverse_solve():

0 commit comments

Comments
 (0)