Use get to avoid KeyError (microsoft#824)

Co-authored-by: Qingyun Wu <[email protected]> Co-authored-by: Kevin Chen <[email protected]>
whiskyboy · Dec 7, 2022 · 00c9875 · 00c9875
1 parent aec19a6
commit 00c9875
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flaml/automl/automl.py b/flaml/automl/automl.py
@@ -235,7 +235,7 @@ def update(self, result, time_used):
             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