Adjust float32 test precision

ricardoV94 · ricardoV94 · commit 30e08e23c1ab · 2023-09-07T11:31:00.000+02:00
diff --git a/tests/tensor/test_blockwise.py b/tests/tensor/test_blockwise.py
@@ -235,7 +235,11 @@ def test_grad(self):
                 )
                 pt_out = pt_func(*vec_inputs_testvals)
                 np_out = np_funcs[test_input_idx](*vec_inputs_testvals)
-                np.testing.assert_allclose(pt_out, np_out, atol=1e-6)
+                np.testing.assert_allclose(
+                    pt_out,
+                    np_out,
+                    atol=1e-6 if config.floatX == "float64" else 1e-5,
+                )
 
 
 class MatrixOpBlockwiseTester(BlockwiseOpTester):
