Fixing #359: decoding str object instead of bytes (#360)

tensorflow · Oct 14, 2017 · 3a9c950 · 3a9c950
1 parent f9f41eb
commit 3a9c950
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensor2tensor/data_generators/cnn_dailymail.py b/tensor2tensor/data_generators/cnn_dailymail.py
@@ -74,7 +74,7 @@ def story_generator(tmp_dir):
   for path in paths:
     for story_file in tf.gfile.Glob(path + "*"):
       story = u""
-      for line in tf.gfile.Open(story_file):
+      for line in tf.gfile.Open(story_file, 'rb'):
         line = unicode(line, "utf-8") if six.PY2 else line.decode("utf-8")
         story += line
       yield story