pytorch_local_mode_cifar10.ipynb failed CI

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

Error:

---------------------------------------------------------------------------
Exception encountered at "In [8]":
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
in
5 framework_version='1.7.1',
6 train_instance_count=1,
----> 7 train_instance_type=instance_type)
8
9 cifar10_estimator.fit(inputs)

/opt/conda/lib/python3.6/site-packages/sagemaker/pytorch/estimator.py in __init__(self, entry_point, framework_version, py_version, source_dir, hyperparameters, image_uri, distribution, **kwargs)
151 :class:`~sagemaker.estimator.EstimatorBase`.
152 """
--> 153 validate_v

[...]

ValueError: framework_version or py_version was None, yet image_uri was also None. Either specify both framework_version and py_version, or specify image_uri.

Open sagemaker-python-sdk/pytorch_cnn_cifar10/pytorch_local_mode_cifar10.ipynb and inspect cell In [8], where the PyTorch estimator is created. Reproduce the CI failure and check the estimator arguments against the reported ValueError. Done means the notebook runs through this step with valid version or image settings and CI passes.

Written by the indexing model from the issue text.