add flaky for failing test

automl · Jul 12, 2022 · e93d766 · e93d766
1 parent e5120df
commit e93d766
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test/test_pipeline/test_tabular_classification.py b/test/test_pipeline/test_tabular_classification.py
@@ -9,6 +9,8 @@
     UniformIntegerHyperparameter,
 )
 
+from flaky import flaky
+
 import numpy as np
 
 import pytest
@@ -59,6 +61,7 @@ def _assert_pipeline_search_space(self, pipeline, search_space_updates):
             elif isinstance(hyperparameter, CategoricalHyperparameter):
                 assert update.value_range == hyperparameter.choices
 
+    @flaky(max_runs=3)
     def test_pipeline_fit(self, fit_dictionary_tabular, exclude):
         """This test makes sure that the pipeline is able to fit
         given random combinations of hyperparameters across the pipeline"""