ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-13 23:28:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix valid model URL might be overridden by an invalid one in case the upload failed

Browse Source
This commit is contained in:
allegroai 2024-08-22 01:26:01 +03:00
parent a9819416fd
commit 0086785372
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clearml/backend_interface/model.py
View File

@ -378,10 +378,13 @@ class Model(IdObjectBase, AsyncManagerMixin, _StorageUriMixin):
if uploaded_uri is None:
return
# If not successful, mark model as failed_uploading
# If not successful, mark model as failed_uploading,
# but dont override valid urls
if uploaded_uri is False:
uploaded_uri = "{}/failed_uploading".format(
self._upload_storage_uri
uploaded_uri = (
self.data.uri
if self.data.uri != "{}/uploading_file".format(self._upload_storage_uri or "file://")
else "{}/failed_uploading".format(self._upload_storage_uri or "file://")
)
Model._local_model_to_id_uri[str(model_file)] = (