infer_pytorch.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/training/distributed_training/pytorch/data_parallel/mnist/infer_pytorch.ipynb

Error:

---------------------------------------------------------------------------
Exception encountered at "In [1]":
---------------------------------------------------------------------------
NameError Traceback (most recent call last)
in
5 # model_data = 's3://sagemaker-us-west-2-XXX/pytorch-smdataparallel-mnist-2020-10-16-17-15-16-419/output/model.tar.gz'
6
----> 7 print("Using this model: {}".format(model_data))

NameError: name 'model_data' is not defined

Open training/distributed_training/pytorch/data_parallel/mnist/infer_pytorch.ipynb and inspect the first cell, where model_data is referenced. Run the notebook or the repository's notebook CI check to reproduce the NameError. Done means the notebook executes past that cell without the reported undefined-variable failure.

Written by the indexing model from the issue text.