From 46d04d7718faf92ad064c2c57faa81a0203d91d3 Mon Sep 17 00:00:00 2001 From: Roman Bredehoft Date: Mon, 18 Sep 2023 13:56:22 +0200 Subject: [PATCH] chore: flag test_torch_padding as flaky --- tests/torch/test_compile_torch.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/torch/test_compile_torch.py b/tests/torch/test_compile_torch.py index be5f50af6..6948e9a52 100644 --- a/tests/torch/test_compile_torch.py +++ b/tests/torch/test_compile_torch.py @@ -1174,6 +1174,9 @@ def test_shape_operations_net( assert "lookup_table" not in quantized_module.fhe_circuit.mlir +# This test is a known flaky, remove the mark once it's fixed +# FIXME: https://github.com/zama-ai/concrete-ml-internal/issues/3989 +@pytest.mark.flaky def test_torch_padding(default_configuration, check_circuit_has_no_tlu): """Test padding in PyTorch using ONNX pad operators.""" net = PaddingNet()