ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 18:52:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix storage_uri handling in Model.update()

Browse Source
This commit is contained in:
allegroai 2020-03-22 18:05:05 +02:00
parent abc9b512f7
commit 477665ee33
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
trains/backend_interface/model.py
View File

@ -106,7 +106,7 @@ class Model(IdObjectBase, AsyncManagerMixin, _StorageUriMixin):
elif res is False: elif res is False:
self.log.info('Failed model upload') self.log.info('Failed model upload')
else: else:
self.log.info('Completed model upload to %s' % res) self.log.info('Completed model upload to {}'.format(res))
if cb: if cb:
cb(res) cb(res)
@ -176,6 +176,8 @@ class Model(IdObjectBase, AsyncManagerMixin, _StorageUriMixin):
if upload_storage_uri: if upload_storage_uri:
self.upload_storage_uri = upload_storage_uri self.upload_storage_uri = upload_storage_uri
self._create_empty_model(self.upload_storage_uri) self._create_empty_model(self.upload_storage_uri)
elif upload_storage_uri:
self.upload_storage_uri = upload_storage_uri
if model_file and uri: if model_file and uri:
Model._local_model_to_id_uri[str(model_file)] = (self.model_id, uri) Model._local_model_to_id_uri[str(model_file)] = (self.model_id, uri)