diff options
author | Annie Tallund <annie.tallund@arm.com> | 2024-02-23 13:05:11 +0100 |
---|---|---|
committer | Annie Tallund <annie.tallund@arm.com> | 2024-03-07 17:30:35 +0100 |
commit | 9a882300efb6049fb7a12d425eda1e43cfbefbe7 (patch) | |
tree | 5a69953b0fa956751c805923cb1d0db02e7b1440 | |
parent | e506c7bd0453cb204ec7a59267fe3982492aaed6 (diff) | |
download | mlia-9a882300efb6049fb7a12d425eda1e43cfbefbe7.tar.gz |
fix: Retry flaky test up to four times
- We have seen flaky tests failing on retrying
- Increasing the number of tries to see if it resolves the issue
Signed-off-by: Annie Tallund <annie.tallund@arm.com>
Change-Id: I15f304a3b7e0e1b15c8e3b8fe22ffacb86f1eda0
-rw-r--r-- | tests/test_nn_tensorflow_optimizations_clustering.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_nn_tensorflow_optimizations_clustering.py b/tests/test_nn_tensorflow_optimizations_clustering.py index 58ffb3e..72ade58 100644 --- a/tests/test_nn_tensorflow_optimizations_clustering.py +++ b/tests/test_nn_tensorflow_optimizations_clustering.py @@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2024, Arm Limited and/or its affiliates. # SPDX-License-Identifier: Apache-2.0 """Test for module optimizations/clustering.""" from __future__ import annotations @@ -86,7 +86,7 @@ def _test_sparsity( # to run the test, and in the meantime we classify it as a known issue. # Additionally, flaky is (as of 2023) untyped and thus we need to silence the # warning from mypy. -@flaky # type: ignore +@flaky(max_runs=4, min_passes=1) # type: ignore @pytest.mark.parametrize("target_num_clusters", (32, 4)) @pytest.mark.parametrize("sparsity_aware", (False, True)) @pytest.mark.parametrize("layers_to_cluster", (["conv1"], ["conv1", "conv2"], None)) |