xgboost-realtime-analysis.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/sagemaker-debugger/xgboost_realtime_analysis/xgboost-realtime-analysis.ipynb

Error:

---------------------------------------------------------------------------
Exception encountered at "In [10]":
---------------------------------------------------------------------------
ModuleNotFoundError Traceback (most recent call last)
in
----> 1 from smdebug.trials import create_trial
2
3 description = client.describe_training_job(TrainingJobName=job_name)
4 s3_output_path = xgboost_algorithm_mode_estimator.latest_job_debugger_artifacts_path()
5

ModuleNotFoundError: No module named 'smdebug'

Open sagemaker-debugger/xgboost_realtime_analysis/xgboost-realtime-analysis.ipynb and start at cell In [10], where the smdebug.trials import fails. Reproduce the notebook’s CI failure and determine what is needed for that import to work. Done means the notebook completes that cell without ModuleNotFoundError.

Written by the indexing model from the issue text.