ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 21:33:25 +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 furl import furl
from pathlib2 import Path from pathlib2 import Path
from requests.exceptions import ConnectionError 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.queue import Queue, Empty
from six.moves.urllib.parse import urlparse from six.moves.urllib.parse import urlparse
from six.moves.urllib.request import url2pathname from six.moves.urllib.request import url2pathname
@ -2084,7 +2084,7 @@ class _FileStorageDriver(_Driver):
self._make_path(base_path) self._make_path(base_path)
obj_path = os.path.realpath(obj_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)) obj_file.write(iterator.read() if hasattr(iterator, 'read') else bytes(iterator))
os.chmod(obj_path, int('664', 8)) os.chmod(obj_path, int('664', 8))