test_multivariate_lmu supports more machines

It needed slightly looser tolerances to pass on my machine.
nengo · Jun 15, 2021 · 8fa5996 · 8fa5996
1 parent edcdf20
commit 8fa5996
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
         )