kge_pytorch_hypertune.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/sagemaker-python-sdk/dgl_kge/kge_pytorch_hypertune.ipynb

Error:

---------------------------------------------------------------------------
Exception encountered at "In [7]":
---------------------------------------------------------------------------
UnexpectedStatusException Traceback (most recent call last)
in
----> 1 tuner.fit()

/opt/conda/lib/python3.6/site-packages/sagemaker/tuner.py in fit(self, inputs, job_name, include_cls_metadata, estimator_kwargs, wait, **kwargs)
447
448 if wait:
--> 449 self.latest_tuning_job.wait()
450
451 def _fit_with_estimator(self, inputs, job_name, include_cls_metadata, **kwargs):

/opt/conda/lib/python3.6/site-packages/sagemaker/tuner.py in wait(self)
1571 def wait(self):
1572 """Placeholder doc

[...]

UnexpectedStatusException: Error for HyperParameterTuning job pytorch-training-210511-0008: Failed. Reason: No training job succeeded after 5 attempts. Please take a look at the training job failures to get more details.