wangtao
/
autogen
mirror of https://github.com/microsoft/autogen.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

time budget in state

This commit is contained in:
Chi Wang 2021-10-16 08:21:14 -07:00
parent b2d8b097d7
commit 9e9356f436
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_training_log.py
View File

@ -39,7 +39,7 @@ class TestTrainingLog(unittest.TestCase):
model0 = automl.best_model_for_estimator(estimator)
print(model0.params["n_estimators"], model0.estimator)
automl.time_budget = None
automl._state.time_budget = None
model, _ = automl._state._train_with_config(estimator, config)
print(model.estimator)
# model0 and model are equivalent unless model0's n_estimator is out of search space range