Increase tolerances on test_multivariate_lmu

nengo · Jun 16, 2021 · 74e9b03 · 74e9b03
1 parent 15d5ba9
commit 74e9b03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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
         )