Skip to content

Commit 1470835

Browse files
set dtype to config.floatX on test quantities
1 parent d8ef2ad commit 1470835

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/tensor/rewriting/test_linalg.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,10 +94,10 @@ def test_tag_solve_triangular():
9494
b2 = solve(U, x)
9595
f = pytensor.function([A, x], b1)
9696

97-
X = np.random.normal(size=(10, 10))
97+
X = np.random.normal(size=(10, 10)).astype(config.floatX)
9898
X = X @ X.T
9999
X_chol = np.linalg.cholesky(X)
100-
eye = np.eye(10)
100+
eye = np.eye(10, dtype=config.floatX)
101101

102102
if config.mode != "FAST_COMPILE":
103103
toposort = f.maker.fgraph.toposort()

0 commit comments

Comments
 (0)