Enable retries when training xgboot on ray

Hello, is there a way to enable retries when training xgboost on ray? I saw a lot of msg as below and I am wondering if enabling retries might fix some of them.

A Ray actor died during training and the maximum number of retries (0) is exhausted

You can configure your XGBoostTrainer with a FailureConfig.