chore: mark test_brevitas_power_of_two as flaky

zama-ai · Apr 4, 2024 · 41673fe · 41673fe
1 parent 5f3c212
commit 41673fe
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/torch/test_brevitas_qat.py b/tests/torch/test_brevitas_qat.py
@@ -495,6 +495,9 @@ def test_brevitas_constant_folding(default_configuration):
         )
 
 
+# This test is a known flaky
+# FIXME: https://github.com/zama-ai/concrete-ml-internal/issues/4356
+@pytest.mark.flaky
 @pytest.mark.parametrize("manual_rounding", [None, 3])
 @pytest.mark.parametrize("power_of_two", [True, False])
 @pytest.mark.parametrize("n_bits", [4])