Skip test_lstm_forward_training_fuzz on aarch64 (#856)

explosion · Mar 1, 2023 · 14cc42a · 14cc42a
1 parent 6d787c4
commit 14cc42a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/thinc/tests/backends/test_ops.py b/thinc/tests/backends/test_ops.py
@@ -2,6 +2,7 @@
 
 import pytest
 import numpy
+import platform
 from hypothesis import given, settings
 from hypothesis.strategies import composite, integers
 from numpy.testing import assert_allclose
@@ -1260,6 +1261,7 @@ def test_lstm_forward_training(ops, depth, dirs, nO, batch_size, nI):
     assert_allclose(Y, reference[0], atol=1e-4, rtol=1e-3)
 
 
+@pytest.mark.skipif(platform.machine() == "aarch64", reason="Flaky, skip temporarily")
 @pytest.mark.parametrize("ops", XP_OPS)
 @settings(max_examples=MAX_EXAMPLES, deadline=None)
 @given(args=draw_lstm_args())