working-with-tfrecords.ipynb failed CI

Link to the notebook:
https://github.com/aws/amazon-sagemaker-examples/blob/master/sagemaker_batch_transform/working_with_tfrecords/working-with-tfrecords.ipynb

Error:

---------------------------------------------------------------------------
Exception encountered at "In [2]":
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
in
21 # create the output file
22 open(tfrecord_root + output_file, 'a').close()
---> 23 with tf.python_io.TFRecordWriter(tfrecord_root + output_file) as writer:
24 with open(csv_root + input_file,'r') as f:
25 f.readline() # skip first line

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

Open sagemaker_batch_transform/working_with_tfrecords/working-with-tfrecords.ipynb and inspect cell In [2], which fails while creating a TFRecord writer. First reproduce the notebook CI failure and check the TensorFlow API available in the test environment. Done means the notebook runs through that cell without the reported AttributeError and CI passes.

Written by the indexing model from the issue text.