MAINT Apply ruff/Perflint rule PERF403

DimitriPapadopoulos · DimitriPapadopoulos · commit e25d49c41b61 · 2025-01-21T23:47:20.000+01:00
PERF403 Use a dictionary comprehension instead of a for-loop
diff --git a/sklearn/model_selection/_search.py b/sklearn/model_selection/_search.py
@@ -1139,8 +1139,12 @@ def _store(key_name, array, weights=None, splits=False, rank=False):
         _store("fit_time", out["fit_time"])
         _store("score_time", out["score_time"])
         # Store a list of param dicts at the key 'params'
-        for param, ma in _yield_masked_array_for_each_param(candidate_params):
-            results[param] = ma
+        results.update(
+            {
+                param: ma
+                for param, ma in _yield_masked_array_for_each_param(candidate_params)
+            }
+        )
         results["params"] = candidate_params
 
         test_scores_dict = _normalize_score_results(out["test_scores"])
