Skip to content

Commit c1567c3

Browse files
committed
fixed failing precision
1 parent 30198d0 commit c1567c3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/tensor/rewriting/test_linalg.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -921,9 +921,9 @@ def test_dot_kron_rewrite():
921921
assert not any(isinstance(node.op, KroneckerProduct) for node in nodes)
922922

923923
# NUMERIC VALUE TEST
924-
a_test = np.random.rand(m, n)
925-
b_test = np.random.rand(p, q)
926-
c_test = np.random.rand(n * q, 1)
924+
a_test = np.random.rand(m, n).astype(config.floatX)
925+
b_test = np.random.rand(p, q).astype(config.floatX)
926+
c_test = np.random.rand(n * q, 1).astype(config.floatX)
927927
out_direct_val = np.kron(a_test, b_test) @ c_test
928928
out_clever_val = f_direct_rewritten(a_test, b_test, c_test)
929929
assert_allclose(

0 commit comments

Comments
 (0)