add return type for xgboost, don't know if it is mac problem