The current pickling function replaced the estimator wrapper with the estimator, which is incorrect as it discards the y variable prepare-step. This may cause failure on the classification problem where the target variable is a string type instead of integer.
Description
The current pickling function replaced the estimator wrapper with the estimator, which is incorrect as it discards the y variable prepare-step. This may cause failure on the classification problem where the target variable is a string type instead of integer.