You have already done this in `_parallel_predict_regression `.