Pulled out callbacks for more customisation.

jameschapman19 · Jun 1, 2022 · ebc5079 · ebc5079
1 parent b87d91b
commit ebc5079
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cca_zoo/model_selection/_search.py b/cca_zoo/model_selection/_search.py
@@ -393,7 +393,7 @@ def fit(self, X, y=None, *, groups=None, **fit_params):
         )
         self = BaseSearchCV.fit(self, np.hstack(X), y=None, groups=None, **fit_params)
         self.best_estimator_ = self.best_estimator_['estimator']
-        self.best_params_ = {key.removeprefix('estimator__'): val for key, val in self.best_params_.items()}
+        self.best_params_ = {key[len('estimator__'):]: val for key, val in self.best_params_.items()}
         return self
 
 
@@ -673,5 +673,5 @@ def fit(self, X, y=None, *, groups=None, **fit_params):
         )
         self = BaseSearchCV.fit(self, np.hstack(X), y=None, groups=None, **fit_params)
         self.best_estimator_ = self.best_estimator_['estimator']
-        self.best_params_ = {key.removeprefix('estimator__'): val for key, val in self.best_params_.items()}
+        self.best_params_ = {key[len('estimator__'):]: val for key, val in self.best_params_.items()}
         return self