tf.python_io should be changed to tf.io

• I am using the latest TensorFlow Model Garden release and TensorFlow 2.
• I am reporting the issue to the correct repository. (Model Garden official or research directory)
• I checked to make sure that this issue has not been filed already.

File in question:

https://github.com/tensorflow/models/blob/master/research/object_detection/dataset_tools/tf_record_creation_util.py

Issue (and proposed fix):

If you run this file, the line from object_detection.dataset_tools import tf_record_creation_util is used, and that file (url above) has the line exit_stack.enter_context(tf.python_io.TFRecordWriter(file_name)), but I believe tf.python_io is depreciated and should be replaced with tf.io.

System information:

tf_env.txt