From b078069643dd082650cf564c66ce410475c54b62 Mon Sep 17 00:00:00 2001 From: Axel Donath Date: Tue, 7 Jan 2025 13:21:56 -0500 Subject: [PATCH] Further decrease test tolerance --- tests/test_gmm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_gmm.py b/tests/test_gmm.py index 0e378af..1814e00 100644 --- 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])