fix: mark test as flaky (#1067)

andrei-stoian-zama · web-flow · commit 2c5f1aba337f · 2025-04-04T18:17:28.000+02:00
diff --git a/tests/sklearn/test_sklearn_models.py b/tests/sklearn/test_sklearn_models.py
@@ -2372,6 +2372,7 @@ def test_xgb_serialization_errors(model_class, param, error_message):
             model.dumps()
 
 
+@pytest.mark.flaky
 @pytest.mark.parametrize(
     "model_class, parameters", get_sklearn_tree_models_and_datasets(True, True)
 )

Original file line number	Diff line number	Diff line change
`@@ -2372,6 +2372,7 @@ def test_xgb_serialization_errors(model_class, param, error_message):`
`2372`	`2372`	`model.dumps()`
`2373`	`2373`
`2374`	`2374`
	`2375`	`+@pytest.mark.flaky`
`2375`	`2376`	`@pytest.mark.parametrize(`
`2376`	`2377`	`"model_class, parameters", get_sklearn_tree_models_and_datasets(True, True)`
`2377`	`2378`	`)`