Skip to content

Commit ae596b6

Browse files
Update tests/tensor/test_slinalg.py
Co-authored-by: Ricardo Vieira <28983449+ricardoV94@users.noreply.github.com>
1 parent fa017de commit ae596b6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tensor/test_slinalg.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1105,7 +1105,7 @@ def test_banded_dot_perf(A_shape, kl, ku, benchmark):
11051105
b = pt.tensor("b", shape=b_val.shape)
11061106

11071107
res = banded_dot(A, b, kl, ku)
1108-
fn = function([A, b], res)
1108+
fn = function([A, b], res, trust_input=True)
11091109

11101110
benchmark(fn, A_val, b_val)
11111111

0 commit comments

Comments
 (0)