Closed justinvyu closed 5 months ago
This PR simplifies the Ray Train/Tune integration for xgboost_ray by doing the following:
xgboost_ray
TuneReportCallback
is_session_enabled
callback._report
callback._checkpoint
TuneReportCheckpointCallback
ray.train.get_context
See the corresponding Ray changes: https://github.com/ray-project/ray/pull/42111
This PR simplifies the Ray Train/Tune integration for
xgboost_rayby doing the following:TuneReportCallback,is_session_enabled,callback._report,callback._checkpoint)TuneReportCheckpointCallbackray.train.get_contextAPI instead of detecting the train "session"See the corresponding Ray changes: https://github.com/ray-project/ray/pull/42111