Updated tolerance in test to account for machine epsilon in float com…

…parison
sm00thix · Jul 7, 2024 · 9ccc4e3 · 9ccc4e3
1 parent f1a60ea
commit 9ccc4e3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_ikpls.py b/tests/test_ikpls.py
@@ -677,7 +677,10 @@ def check_cpu_gpu_equality(
         if n_good_components == -1:
             n_good_components = np_pls_alg_1.A
 
-        atol = 0
+        try:
+            atol = np.finfo(np_pls_alg_1.dtype).eps
+        except AttributeError:
+            atol = 0
         rtol = 1e-4
         # Regression matrices
         assert_allclose(