Fixed tf.compat.v1 bug

a5c96853 · Mirali Ahmadli · 0effd158 · a5c96853
Commit a5c96853 authored Nov 29, 2019 by Mirali Ahmadli
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)