remove check for fit_predict