Move references to `tf_record_iterator`. (#5830)

For tf2 this will only be available in `compat.v1`.

Move references to `tf_record_iterator`. (#5830)
For tf2 this will only be available in `compat.v1`.
a1eb92b0 · Mark Daoust · Taylor Robie · aead3912 · a1eb92b0
Commit a1eb92b0 authored Dec 19, 2018 by Mark Daoust Committed by Taylor Robie Dec 19, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/transformer/data_download.py official/transformer/data_download.py +1 -1

No files found.
--- a/official/transformer/data_download.py
+++ b/official/transformer/data_download.py
@@ -321,7 +321,7 @@ def shuffle_records(fname):
  tmp_fname = fname + ".unshuffled"
  tf.gfile.Rename(fname, tmp_fname)

-  reader = tf.python_io.tf_record_iterator(tmp_fname)
+  reader = tf.compat.v1.io.tf_record_iterator(tmp_fname)
  records = []
  for record in reader:
    records.append(record)