Increase flaky test count

borchero · Dec 21, 2021 · 30e0fdf · 30e0fdf
1 parent fed2f31
commit 30e0fdf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"),
     [