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

Fix when abort callback is set, set task status to stopped only if running locally, otherwise leave it for the Agent to set it.

Browse Source
This commit is contained in:
allegroai 2024-07-31 17:25:12 +03:00
parent b298e212dd
commit 62dd92a22d
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
clearml/task.py
View File

@ -4148,7 +4148,10 @@ class Task(_Task):
)
)
self.flush(wait_for_uploads=True)
self.stopped(status_reason='USER ABORTED')
# if running remotely, we want the daemon to kill us
if self.running_locally():
self.stopped(status_reason='USER ABORTED')
if self._dev_worker:
self._dev_worker.unregister()