WIP: kernel mmv tests failing, maybe too much precision required

FalkonML · Oct 17, 2023 · 38061eb · 38061eb
1 parent f88b9ba
commit 38061eb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/falkon/tests/test_kernels.py b/falkon/tests/test_kernels.py
@@ -73,12 +73,12 @@ def w() -> torch.Tensor:
 
 @pytest.fixture(scope="module")
 def rtol():
-    return {np.float64: 1e-12, torch.float64: 1e-12, np.float32: 1e-4, torch.float32: 1e-4}
+    return {np.float64: 1e-10, torch.float64: 1e-10, np.float32: 1e-4, torch.float32: 1e-4}
 
 
 @pytest.fixture(scope="module")
 def atol():
-    return {np.float64: 1e-12, torch.float64: 1e-12, np.float32: 1e-4, torch.float32: 1e-4}
+    return {np.float64: 1e-10, torch.float64: 1e-10, np.float32: 1e-4, torch.float32: 1e-4}
 
 
 @pytest.fixture(params=["single-sigma", "vec-sigma"], scope="class")