Merge pull request #7884 from miraliahmadli:master

PiperOrigin-RevId: 283114824

Merge pull request #7884 from miraliahmadli:master
PiperOrigin-RevId: 283114824
676c23e6 · A. Unique TensorFlower · d49cb9ef · a5c96853 · 676c23e6
Commit 676c23e6 authored Nov 29, 2019 by A. Unique TensorFlower
Show 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
@@ -331,7 +331,7 @@ def shuffle_records(fname):
  tmp_fname = fname + ".unshuffled"
  tf.gfile.Rename(fname, tmp_fname)

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