Fix KNeighborsRegressor returning NaN predictions