You've lost the lines calling the estimator without y_train when y was None. Do we have a clear view of the consequences? If not, I'd rather keep them.