diff --git a/n3fit/src/n3fit/tests/test_multidense.py b/n3fit/src/n3fit/tests/test_multidense.py index 14bbdb9817..ed848ffedc 100644 --- a/n3fit/src/n3fit/tests/test_multidense.py +++ b/n3fit/src/n3fit/tests/test_multidense.py @@ -42,7 +42,7 @@ def test_multidense(): [single_model(test_input) for single_model in single_models], axis=1 ) - np.testing.assert_allclose(multi_dense_output, single_dense_output) + np.testing.assert_allclose(multi_dense_output, single_dense_output, atol=1e-6, rtol=1e-4) def test_initializers():