Increase flaky test count

borchero · borchero · commit 30e0fdfd8386 · 2021-12-21T03:28:41.000+01:00
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"),
     [

Original file line number	Diff line number	Diff line change
`@@ -46,7 +46,7 @@ def test_fit_converged(num_epochs: int, converged: bool):`
`46`	`46`	`assert estimator.converged_ == converged`
`47`	`47`
`48`	`48`
`49`		`-@pytest.mark.flaky(max_runs=2, min_passes=1)`
	`49`	`+@pytest.mark.flaky(max_runs=5, min_passes=1)`
`50`	`50`	`@pytest.mark.parametrize(`
`51`	`51`	`("num_datapoints", "batch_size", "num_features", "num_centroids"),`
`52`	`52`	`[`