Skip to content

Commit

Permalink
unit test max time updated due to slow github actions environment
Browse files Browse the repository at this point in the history
  • Loading branch information
nickmatsumoto committed Jul 28, 2023
1 parent 21af4c4 commit 22032bd
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tpot2/tests/test_estimators.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,15 +7,15 @@
#standard test
@pytest.fixture
def tpot_estimator():
return tpot2.TPOTEstimator( population_size=10,
return tpot2.TPOTEstimator( population_size=50,
scorers=['roc_auc_ovr'],
scorers_weights=[1],
classification=True,
n_jobs=1,
early_stop=5,
other_objective_functions= [],
other_objective_functions_weights=[],
max_time_seconds=30,
max_time_seconds=90,
verbose=1)

@pytest.fixture
Expand Down Expand Up @@ -81,11 +81,11 @@ def test_tpot_estimator_config_dict_type():

@pytest.fixture
def tpot_classifier():
return tpot2.tpot_estimator.templates.TPOTClassifier(max_time_seconds=30,verbose=1)
return tpot2.tpot_estimator.templates.TPOTClassifier(max_time_seconds=90,verbose=1)

@pytest.fixture
def tpot_regressor():
return tpot2.tpot_estimator.templates.TPOTRegressor(max_time_seconds=60,verbose=1)
return tpot2.tpot_estimator.templates.TPOTRegressor(max_time_seconds=90,verbose=1)

def test_tpot_classifier_fit(tpot_classifier,sample_dataset):
#load iris dataset
Expand Down

0 comments on commit 22032bd

Please sign in to comment.