diff options
Diffstat (limited to 'tensorflow/tensorboard/backend/handler.py')
| -rw-r--r-- | tensorflow/tensorboard/backend/handler.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/tensorboard/backend/handler.py b/tensorflow/tensorboard/backend/handler.py index a8ecf73a5b..1c1e9411a4 100644 --- a/tensorflow/tensorboard/backend/handler.py +++ b/tensorflow/tensorboard/backend/handler.py @@ -32,6 +32,7 @@ import os import re from six import BytesIO +from six import StringIO from six.moves import BaseHTTPServer from six.moves import urllib from six.moves import xrange # pylint: disable=redefined-builtin @@ -276,7 +277,7 @@ class TensorboardHandler(BaseHTTPServer.BaseHTTPRequestHandler): values = self._multiplexer.Scalars(run, tag) if query_params.get('format') == _OutputFormat.CSV: - string_io = BytesIO() + string_io = StringIO() writer = csv.writer(string_io) writer.writerow(['Wall time', 'Step', 'Value']) writer.writerows(values) @@ -353,7 +354,7 @@ class TensorboardHandler(BaseHTTPServer.BaseHTTPRequestHandler): run = query_params.get('run') compressed_histograms = self._multiplexer.CompressedHistograms(run, tag) if query_params.get('format') == _OutputFormat.CSV: - string_io = BytesIO() + string_io = StringIO() writer = csv.writer(string_io) # Build the headers; we have two columns for timing and two columns for |