diff --git a/tpot2/tests/test_estimators.py b/tpot2/tests/test_estimators.py index a60b867f..8a46b74a 100644 --- 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