added atol to output comparison

infer-actively · Jun 6, 2024 · 51c799f · 51c799f
1 parent a7d5a29
commit 51c799f
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
@@ -143,7 +143,7 @@ def test_fixed_point_iteration_factorized_fullyconnected(self):
             qs_jax_factorized = fpi_jax_factorized(A, obs, prior, factor_lists, num_iter=16)
 
             for f, _ in enumerate(qs_jax):
-                self.assertTrue(np.allclose(qs_jax[f], qs_jax_factorized[f]))
+                self.assertTrue(np.allclose(qs_jax[f], qs_jax_factorized[f], atol=1e-6))
 
     def test_fixed_point_iteration_factorized_sparsegraph(self):
         """