ClearML/clearml-session
1
0
Fork 0
mirror of https://github.com/clearml/clearml-session synced 2025-06-26 18:16:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix --upload-files should wait until artifact is fully uploaded

Browse Source
This commit is contained in:
allegroai 2023-10-09 13:51:01 +03:00
parent 5a7423069c
commit 4ad0e30091
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
clearml_session/__main__.py
View File

@ -204,7 +204,13 @@ def create_base_task(state, project_name=None, task_name=None):
# if we need to upload data now is the time # if we need to upload data now is the time
if state.get("upload_files"): if state.get("upload_files"):
task.upload_artifact(name="session-files", artifact_object=Path(state.get("upload_files")).expanduser()) print("Uploading local files: {}".format(state.get("upload_files")))
task.upload_artifact(
name="session-files",
artifact_object=Path(state.get("upload_files")).expanduser(),
wait_on_upload=True
)
print("Local files upload completed")
# only update the data at the end, so reload requests are smaller # only update the data at the end, so reload requests are smaller
# noinspection PyProtectedMember # noinspection PyProtectedMember