From dbc2e2d7963c36ec6111c22a79476ad522dd04ec Mon Sep 17 00:00:00 2001 From: Chi Wang Date: Tue, 6 Dec 2022 22:23:45 -0800 Subject: [PATCH] Use get to avoid KeyError (#824) Co-authored-by: Qingyun Wu Co-authored-by: Kevin Chen <74878789+int-chaos@users.noreply.github.com> --- flaml/automl/automl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flaml/automl/automl.py b/flaml/automl/automl.py index 794bb1748..0990172c5 100644 --- a/flaml/automl/automl.py +++ b/flaml/automl/automl.py @@ -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