Fixes column name

LudvigOlsen · Aug 26, 2024 · 911f0bf · 911f0bf
1 parent 89f779c
commit 911f0bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/generalize/model/cross_validate/grid_search.py b/generalize/model/cross_validate/grid_search.py
@@ -573,7 +573,7 @@ def simplest_model_refit_strategy(cv_results):
                     )
                 var_col_nm, var_key_nm = var_split
 
-                cv_results.loc[:, var_col_nm] = cv_results["params"].apply(
+                cv_results.loc[:, var_nm] = cv_results["params"].apply(
                     lambda d: d[var_col_nm][var_key_nm.lstrip("_")]
                 )