From 74e9b033d8369f814c9b7e63195e23e4206fbbb7 Mon Sep 17 00:00:00 2001 From: Eric Hunsberger Date: Tue, 15 Jun 2021 15:57:54 -0400 Subject: [PATCH] Increase tolerances on test_multivariate_lmu --- keras_lmu/tests/test_layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_lmu/tests/test_layers.py b/keras_lmu/tests/test_layers.py index 2865e7ef..967ff178 100644 --- a/keras_lmu/tests/test_layers.py +++ b/keras_lmu/tests/test_layers.py @@ -58,7 +58,7 @@ def test_multivariate_lmu(rng): for i in range(memory_d): assert np.allclose( - results[0][..., i * order : (i + 1) * order], results[i + 1], atol=1e-6 + results[0][..., i * order : (i + 1) * order], results[i + 1], atol=2e-6 )