reduce assertion absolute tollerance

infer-actively · Jun 17, 2024 · 62a6e87 · 62a6e87
1 parent bfc1346
commit 62a6e87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_message_passing_jax.py b/test/test_message_passing_jax.py
@@ -108,7 +108,7 @@ def test_fixed_point_iteration(self):
             qs_jax = fpi_jax(A, obs, prior, num_iter=16)
 
             for f, _ in enumerate(qs_jax):
-                self.assertTrue(np.allclose(qs_numpy[f], qs_jax[f]))
+                self.assertTrue(np.allclose(qs_numpy[f], qs_jax[f], atol=1e-6))
 
 
     def test_fixed_point_iteration_factorized_fullyconnected(self):