04c_pytorch_training.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/prep_data/image_data_guide/04c_pytorch_training.ipynb

Error:

---------------------------------------------------------------------------
Exception encountered at "In [2]":
---------------------------------------------------------------------------
FileNotFoundError Traceback (most recent call last)
in
----> 1 with open('pickled_data/category_labels.pickle', 'rb') as f:
2 category_labels = pickle.load(f)
3
4 with open('pickled_data/pytorch_bucket_name.pickle', 'rb') as f:
5 bucket_name = pickle.load(f)

FileNotFoundError: [Errno 2] No such file or directory: 'pickled_data/category_labels.pickle'

Open prep_data/image_data_guide/04c_pytorch_training.ipynb and reproduce the CI failure at In [2]. Trace the notebook's setup around pickled_data/category_labels.pickle and pickled_data/pytorch_bucket_name.pickle, then verify the notebook completes CI without the reported FileNotFoundError.

Written by the indexing model from the issue text.