Remove global ATOL/RTOL

Author: jessegrabowski

tests/link/numba/test_slinalg.py

@@ -18,8 +18,6 @@
 
 floatX = pytensor.config.floatX
 
-ATOL = 1e-8 if floatX.endswith("64") else 1e-4
-RTOL = 1e-8 if floatX.endswith("64") else 1e-4
 rng = np.random.default_rng(42849)
 
 
@@ -84,6 +82,10 @@ def A_func(x):
     X_np = f(A_func(A_val.copy()), b_val.copy())
 
     test_input = transpose_func(A_func(A_val.copy()), trans)
+
+    ATOL = 1e-8 if floatX.endswith("64") else 1e-4
+    RTOL = 1e-8 if floatX.endswith("64") else 1e-4
+
     np.testing.assert_allclose(test_input @ X_np, b_val, atol=ATOL, rtol=RTOL)
 
     compare_numba_and_py(f.maker.fgraph, [A_func(A_val.copy()), b_val.copy()])
@@ -403,6 +405,9 @@ def A_func(x):
     assert (A_val == A_val_copy).all() == (op.destroy_map.get(0, None) != [0])
     assert (b_val == b_val_copy).all() == (op.destroy_map.get(0, None) != [1])
 
+    ATOL = 1e-8 if floatX.endswith("64") else 1e-4
+    RTOL = 1e-8 if floatX.endswith("64") else 1e-4
+
     # Confirm b_val is used to store to solution
     np.testing.assert_allclose(X_np, b_val, atol=ATOL, rtol=RTOL)
     assert not np.allclose(b_val, b_val_copy)
@@ -444,4 +449,7 @@ def test_cho_solve(b_func, b_size, lower):
 
     X_np = f(A, b)
 
+    ATOL = 1e-8 if floatX.endswith("64") else 1e-4
+    RTOL = 1e-8 if floatX.endswith("64") else 1e-4
+
     np.testing.assert_allclose(A @ X_np, b, atol=ATOL, rtol=RTOL)

tests/tensor/test_slinalg.py

@@ -31,10 +31,6 @@
 from tests import unittest_tools as utt
 
 
-ATOL = 1e-8 if config.floatX.endswith("64") else 1e-4
-RTOL = 1e-8 if config.floatX.endswith("64") else 1e-4
-
-
 def check_lower_triangular(pd, ch_f):
     ch = ch_f(pd)
     assert ch[0, pd.shape[1] - 1] == 0
@@ -265,6 +261,9 @@ def A_func(x):
     solve_func = pytensor.function([A, b], y)
     X_np = solve_func(A_val.copy(), b_val.copy())
 
+    ATOL = 1e-8 if config.floatX.endswith("64") else 1e-4
+    RTOL = 1e-8 if config.floatX.endswith("64") else 1e-4
+
     np.testing.assert_allclose(
         scipy.linalg.solve(solve_input_val, b_val, assume_a=assume_a),
         X_np,