ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 17:17:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix preview with JSON (dict) artifacts did not store the artifact

Browse Source
This commit is contained in:
allegroai 2020-11-17 23:09:08 +02:00
parent 909f5cea2d
commit b656e0115f
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
trains/binding/artifacts.py
View File

@ -376,12 +376,13 @@ class Artifacts(object):
elif isinstance(artifact_object, dict): elif isinstance(artifact_object, dict):
artifact_type = 'JSON' artifact_type = 'JSON'
artifact_type_data.content_type = 'application/json' artifact_type_data.content_type = 'application/json'
preview = preview or json.dumps(artifact_object, sort_keys=True, indent=4) json_text = json.dumps(artifact_object, sort_keys=True, indent=4)
override_filename_ext_in_uri = '.json' override_filename_ext_in_uri = '.json'
override_filename_in_uri = name + override_filename_ext_in_uri override_filename_in_uri = name + override_filename_ext_in_uri
fd, local_filename = mkstemp(prefix=quote(name, safe="") + '.', suffix=override_filename_ext_in_uri) fd, local_filename = mkstemp(prefix=quote(name, safe="") + '.', suffix=override_filename_ext_in_uri)
os.write(fd, bytes(preview.encode())) os.write(fd, bytes(json_text.encode()))
os.close(fd) os.close(fd)
preview = preview or json_text
if len(preview) < self.max_preview_size_bytes: if len(preview) < self.max_preview_size_bytes:
artifact_type_data.preview = preview artifact_type_data.preview = preview
else: else: