From b8ff29d7d5a3a3075b0b7034383228b406fe579c Mon Sep 17 00:00:00 2001 From: Roman Bredehoft Date: Thu, 4 Apr 2024 11:41:22 +0200 Subject: [PATCH] chore: mark test_brevitas_power_of_two as flaky --- tests/torch/test_brevitas_qat.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/torch/test_brevitas_qat.py b/tests/torch/test_brevitas_qat.py index fa778431a..cf6f70c7e 100644 --- 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])