diff options
Diffstat (limited to 'tensorflow/tensorboard/backend/server_test.py')
| -rw-r--r-- | tensorflow/tensorboard/backend/server_test.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/tensorboard/backend/server_test.py b/tensorflow/tensorboard/backend/server_test.py index aceaced26c..42c2aafe21 100644 --- a/tensorflow/tensorboard/backend/server_test.py +++ b/tensorflow/tensorboard/backend/server_test.py @@ -26,10 +26,10 @@ import gzip import json import os import shutil -import StringIO import threading import zlib +from six import BytesIO from six.moves import http_client from six.moves import xrange # pylint: disable=redefined-builtin import tensorflow as tf @@ -80,9 +80,9 @@ class TensorboardServerTest(tf.test.TestCase): content = response.read() if encoding in ('gzip', 'x-gzip', 'deflate'): if encoding == 'deflate': - data = StringIO.StringIO(zlib.decompress(content)) + data = BytesIO(zlib.decompress(content)) else: - data = gzip.GzipFile('', 'rb', 9, StringIO.StringIO(content)) + data = gzip.GzipFile('', 'rb', 9, BytesIO(content)) content = data.read() return content @@ -201,7 +201,7 @@ class TensorboardServerTest(tf.test.TestCase): node1.name = 'a' node2 = graph_def.node.add() node2.name = 'b' - node2.attr['very_large_attr'].s = 'a' * 2048 # 2 KB attribute + node2.attr['very_large_attr'].s = b'a' * 2048 # 2 KB attribute writer.add_event(tf.Event(graph_def=graph_def.SerializeToString())) # 1x1 transparent GIF. |