Further decrease test tolerance

adonath · Jan 7, 2025 · b078069 · b078069
1 parent b6b40db
commit b078069
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_gmm.py b/tests/test_gmm.py
@@ -214,7 +214,7 @@ def test_sklearn_api(gmm_jax):
 
     value = gmm.predict_proba(x[:2])
     assert_allclose(
-        value, [[9.990787e-01, 9.212347e-04], [5.027220e-01, 4.972780e-01]], atol=1e-4
+        value, [[9.990787e-01, 9.212347e-04], [5.027220e-01, 4.972780e-01]], atol=1e-3
     )
 
     value = gmm.aic(x[:2])