From 30e0fdfd83863f972029711bb43180146817c8fe Mon Sep 17 00:00:00 2001 From: Oliver Borchert Date: Tue, 21 Dec 2021 03:28:41 +0100 Subject: [PATCH] Increase flaky test count --- tests/clustering/kmeans/test_kmeans_estimator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/clustering/kmeans/test_kmeans_estimator.py b/tests/clustering/kmeans/test_kmeans_estimator.py index abfba79..0e2bc76 100644 --- a/tests/clustering/kmeans/test_kmeans_estimator.py +++ b/tests/clustering/kmeans/test_kmeans_estimator.py @@ -46,7 +46,7 @@ def test_fit_converged(num_epochs: int, converged: bool): assert estimator.converged_ == converged -@pytest.mark.flaky(max_runs=2, min_passes=1) +@pytest.mark.flaky(max_runs=5, min_passes=1) @pytest.mark.parametrize( ("num_datapoints", "batch_size", "num_features", "num_centroids"), [