From 477665ee335e74152888457427553d28927dc871 Mon Sep 17 00:00:00 2001 From: allegroai <> Date: Sun, 22 Mar 2020 18:05:05 +0200 Subject: [PATCH] Fix storage_uri handling in Model.update() --- trains/backend_interface/model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/trains/backend_interface/model.py b/trains/backend_interface/model.py index 2d4c3ece..c48b9360 100644 --- a/trains/backend_interface/model.py +++ b/trains/backend_interface/model.py @@ -106,7 +106,7 @@ class Model(IdObjectBase, AsyncManagerMixin, _StorageUriMixin): elif res is False: self.log.info('Failed model upload') else: - self.log.info('Completed model upload to %s' % res) + self.log.info('Completed model upload to {}'.format(res)) if cb: cb(res) @@ -176,6 +176,8 @@ class Model(IdObjectBase, AsyncManagerMixin, _StorageUriMixin): if upload_storage_uri: self.upload_storage_uri = 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: Model._local_model_to_id_uri[str(model_file)] = (self.model_id, uri)