Skip to content

Commit

Permalink
chore: add test_reduce_sum configuration as flaky
Browse files Browse the repository at this point in the history
  • Loading branch information
RomanBredehoft committed Apr 4, 2024
1 parent c1f4b78 commit 9385d64
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions tests/torch/test_reduce_sum.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,9 @@
from concrete.ml.torch.compile import compile_torch_model


# This test is a known flaky test
# FIXME: https://github.com/zama-ai/concrete-ml-internal/issues/4357
@pytest.mark.flaky
@pytest.mark.parametrize(
"data_generator",
[
Expand Down Expand Up @@ -65,13 +68,9 @@ def test_sum(
check_circuit_has_no_tlu,
check_circuit_precision,
check_r2_score,
is_weekly_option,
):
"""Tests ReduceSum ONNX operator on a torch model."""

if with_pbs and not is_weekly_option:
pytest.skip("Tests on model with some PBS take too long for regular CIs")

# Generate the input-set with several samples. This adds a necessary batch size
inputset = data_generator(size=(100,) + size)

Expand Down

0 comments on commit 9385d64

Please sign in to comment.