github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-01-25 03:58:09 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use get to avoid KeyError (#824)

Browse Source 
Co-authored-by: Qingyun Wu <qingyun.wu@psu.edu>
Co-authored-by: Kevin Chen <74878789+int-chaos@users.noreply.github.com>
This commit is contained in:
Chi Wang
2022-12-06 22:23:45 -08:00
committed by GitHub
parent f8fa5258e1
commit dbc2e2d796
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
flaml/automl/automl.py
View File

@@ -235,7 +235,7 @@ class SearchState:
n_iter = (
trained_estimator
and hasattr(trained_estimator, "ITER_HP")
and trained_estimator.params[trained_estimator.ITER_HP]
and trained_estimator.params.get(trained_estimator.ITER_HP)
)
if n_iter:
config[trained_estimator.ITER_HP] = n_iter