[Bug Report] Rule - class_imbalance on XGBoost

I am facing this error while running class_imbalance on XGboost

'RuleConfigurationName': 'ClassImbalance',
'RuleEvaluationStatus': 'Error',
'StatusDetails': 'ClientError: No debugging data was saved by the training job. Check that the debugger hook was configured correctly before starting the training job. Exception: Training job has ended. All the collection files could not be loaded\nTraceback (most recent call last):\n File "evaluate.py" .......

I have added below mentioned collections in the DebuggerHookConfig
metrics
predictions
labels.

Can you suggest me what I am missing over here?

No notebook, test, or source file is named. Start by reproducing the XGBoost class_imbalance run and inspect the DebuggerHookConfig collections for metrics, predictions, and labels. Done means the training job saves the collection files and the ClassImbalance rule completes without the reported ClientError.

Written by the indexing model from the issue text.