tensorflow_moving_from_framework_mode_to_script_mode.ipynb failed CI

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

Error:

---------------------------------------------------------------------------
Exception encountered at "In [2]":
---------------------------------------------------------------------------
ModuleNotFoundError Traceback (most recent call last)
in
1 import utils
----> 2 from tensorflow.examples.tutorials.mnist import input_data
3 import tensorflow as tf
4
5 data_sets = input_data.read_data_sets('data', dtype=tf.uint8, reshape=False, validation_size=5000)

ModuleNotFoundError: No module named 'tensorflow.examples'

Open sagemaker-python-sdk/tensorflow_moving_from_framework_mode_to_script_mode/tensorflow_moving_from_framework_mode_to_script_mode.ipynb and inspect the failing In[2] cell, especially the TensorFlow MNIST import. Run the notebook or its CI check to reproduce the ModuleNotFoundError; done means the notebook completes that step without the missing tensorflow.examples module error.

Written by the indexing model from the issue text.