tensorflow_BYOM_iris.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/advanced_functionality/tensorflow_iris_byom/tensorflow_BYOM_iris.ipynb

Error:

---------------------------------------------------------------------------
Exception encountered at "In [5]":
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
in
1 from iris_dnn_classifier import train_input_fn
----> 2 train_func = train_input_fn('.', params = None)

/opt/ml/processing/input/iris_dnn_classifier.py in train_input_fn(training_dir, params)
18 def train_input_fn(training_dir, params):
19 """Returns input function that would feed the model during training"""
---> 20 return _generate_input_fn(training_dir, 'iris_training.csv')
21
22 def _generate_input_fn(training_dir, training_filename):

/opt/ml/processing/input/iris_dnn_classifier.py in _generate_input_fn(training_dir, training_filename)
21
22 def _generate_input_fn(training_dir, trainin

[...]

AttributeError: module 'tensorflow' has no attribute 'contrib'

Start with advanced_functionality/tensorflow_iris_byom/tensorflow_BYOM_iris.ipynb and the referenced iris_dnn_classifier.py, focusing on the failing In [5] cell and its train_input_fn traceback. Run the notebook's CI check to reproduce the TensorFlow AttributeError, then confirm the notebook completes successfully without the failure.

Written by the indexing model from the issue text.