diff options
author | Andrew Selle <aselle@google.com> | 2017-01-11 19:25:21 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-11 19:45:51 -0800 |
commit | ded01d2902355c0ae0c5e70b300e32fda0911392 (patch) | |
tree | dcf31407242c64b7a40295e6376d2e41b125baa0 /tensorflow/tools/compatibility | |
parent | 2bdf0b4e199ad18db41412a4164affe5d6d7b06f (diff) |
Python3 Compatibility for tf_upgrade_test.py
Change: 144279869
Diffstat (limited to 'tensorflow/tools/compatibility')
-rw-r--r-- | tensorflow/tools/compatibility/tf_upgrade_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/tools/compatibility/tf_upgrade_test.py b/tensorflow/tools/compatibility/tf_upgrade_test.py index 69a85d8bda..7548b38b91 100644 --- a/tensorflow/tools/compatibility/tf_upgrade_test.py +++ b/tensorflow/tools/compatibility/tf_upgrade_test.py @@ -17,7 +17,7 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function -import StringIO +import six from tensorflow.python.framework import test_util from tensorflow.python.platform import test as test_lib from tensorflow.tools.compatibility import tf_upgrade @@ -32,8 +32,8 @@ class TestUpgrade(test_util.TensorFlowTestCase): """ def _upgrade(self, old_file_text): - in_file = StringIO.StringIO(old_file_text) - out_file = StringIO.StringIO() + in_file = six.StringIO(old_file_text) + out_file = six.StringIO() upgrader = tf_upgrade.TensorFlowCodeUpgrader() count, report, errors = ( upgrader.process_opened_file("test.py", in_file, |