From ded01d2902355c0ae0c5e70b300e32fda0911392 Mon Sep 17 00:00:00 2001 From: Andrew Selle Date: Wed, 11 Jan 2017 19:25:21 -0800 Subject: [PATCH] Python3 Compatibility for tf_upgrade_test.py Change: 144279869 --- tensorflow/tools/compatibility/tf_upgrade_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tensorflow/tools/compatibility/tf_upgrade_test.py b/tensorflow/tools/compatibility/tf_upgrade_test.py index 69a85d8bda9..7548b38b918 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,