updating the max time for the unit tests

EpistasisLab · Jul 28, 2023 · e646215 · e646215
1 parent 5b977c6
commit e646215
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tpot2/tests/test_estimators.py b/tpot2/tests/test_estimators.py
@@ -7,7 +7,7 @@
 #standard test
 @pytest.fixture
 def tpot_estimator():
-    return tpot2.TPOTEstimator(  population_size=2,
+    return tpot2.TPOTEstimator(  population_size=10,
                             generations=2,
                             scorers=['roc_auc_ovr'],
                             scorers_weights=[1],
@@ -16,6 +16,7 @@ def tpot_estimator():
                             early_stop=5,
                             other_objective_functions= [],
                             other_objective_functions_weights=[],
+                            max_time_seconds=30,
                             verbose=1)
 
 @pytest.fixture
@@ -81,11 +82,11 @@ def test_tpot_estimator_config_dict_type():
 
 @pytest.fixture
 def tpot_classifier():
-    return tpot2.tpot_estimator.templates.TPOTClassifier(max_time_seconds=10,verbose=1)
+    return tpot2.tpot_estimator.templates.TPOTClassifier(max_time_seconds=30,verbose=1)
 
 @pytest.fixture
 def tpot_regressor():
-    return tpot2.tpot_estimator.templates.TPOTRegressor(max_time_seconds=10,verbose=1)
+    return tpot2.tpot_estimator.templates.TPOTRegressor(max_time_seconds=30,verbose=1)
 
 def test_tpot_classifier_fit(tpot_classifier,sample_dataset):
     #load iris dataset