fix tests on float32

dehorsley · dehorsley · commit 138daee0ce16 · 2023-05-20T08:59:26.000+10:00
diff --git a/tests/tensor/rewriting/test_linalg.py b/tests/tensor/rewriting/test_linalg.py
@@ -150,15 +150,16 @@ def test_cholesky_ldotlt(tag, cholesky_form, product):
     # Test some concrete value through f
     # there must be lower triangular (f assumes they are)
     Avs = [
-        np.eye(1),
-        np.eye(10),
-        np.array([[2, 0], [1, 4]]),
+        np.eye(1, dtype=pytensor.config.floatX),
+        np.eye(10, dtype=pytensor.config.floatX),
+        np.array([[2, 0], [1, 4]], dtype=pytensor.config.floatX),
     ]
     if not tag:
         # these must be positive def
         Avs.extend(
             [
-                np.ones((4, 4)) + np.eye(4),
+                np.ones((4, 4), dtype=pytensor.config.floatX)
+                + np.eye(4, dtype=pytensor.config.floatX),
             ]
         )
 

Original file line number	Diff line number	Diff line change
`@@ -150,15 +150,16 @@ def test_cholesky_ldotlt(tag, cholesky_form, product):`
`150`	`150`	`# Test some concrete value through f`
`151`	`151`	`# there must be lower triangular (f assumes they are)`
`152`	`152`	`Avs = [`
`153`		`- np.eye(1),`
`154`		`- np.eye(10),`
`155`		`- np.array([[2, 0], [1, 4]]),`
	`153`	`+ np.eye(1, dtype=pytensor.config.floatX),`
	`154`	`+ np.eye(10, dtype=pytensor.config.floatX),`
	`155`	`+ np.array([[2, 0], [1, 4]], dtype=pytensor.config.floatX),`
`156`	`156`	`]`
`157`	`157`	`if not tag:`
`158`	`158`	`# these must be positive def`
`159`	`159`	`Avs.extend(`
`160`	`160`	`[`
`161`		`- np.ones((4, 4)) + np.eye(4),`
	`161`	`+ np.ones((4, 4), dtype=pytensor.config.floatX)`
	`162`	`+ + np.eye(4, dtype=pytensor.config.floatX),`
`162`	`163`	`]`
`163`	`164`	`)`
`164`	`165`