regressor time update for unit test

EpistasisLab · Jul 28, 2023 · 21af4c4 · 21af4c4
1 parent e646215
commit 21af4c4
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tpot2/tests/test_estimators.py b/tpot2/tests/test_estimators.py
@@ -8,7 +8,6 @@
 @pytest.fixture
 def tpot_estimator():
     return tpot2.TPOTEstimator(  population_size=10,
-                            generations=2,
                             scorers=['roc_auc_ovr'],
                             scorers_weights=[1],
                             classification=True,
@@ -86,7 +85,7 @@ def tpot_classifier():
 
 @pytest.fixture
 def tpot_regressor():
-    return tpot2.tpot_estimator.templates.TPOTRegressor(max_time_seconds=30,verbose=1)
+    return tpot2.tpot_estimator.templates.TPOTRegressor(max_time_seconds=60,verbose=1)
 
 def test_tpot_classifier_fit(tpot_classifier,sample_dataset):
     #load iris dataset