ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix StorageHelper to stream StringIO into file (tensorboard add_text support)

Browse Source
This commit is contained in:
allegroai 2020-07-30 14:57:35 +03:00
parent 7de064aaa0
commit d8671beb89
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
trains/storage/helper.py
View File

@ -25,7 +25,7 @@ from attr import attrs, attrib, asdict
from furl import furl
from pathlib2 import Path
from requests.exceptions import ConnectionError
from six import binary_type
from six import binary_type, StringIO
from six.moves.queue import Queue, Empty
from six.moves.urllib.parse import urlparse
from six.moves.urllib.request import url2pathname
@ -2084,7 +2084,7 @@ class _FileStorageDriver(_Driver):
self._make_path(base_path)
obj_path = os.path.realpath(obj_path)
with open(obj_path, 'wb') as obj_file:
with open(obj_path, 'wb' if not isinstance(iterator, StringIO) else 'wt') as obj_file:
obj_file.write(iterator.read() if hasattr(iterator, 'read') else bytes(iterator))
os.chmod(obj_path, int('664', 8))