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

Fix queued task is not dequeued on tasks.stop

Browse Source
This commit is contained in:
allegroai 2021-07-25 14:32:09 +03:00
parent 6705aff56f
commit 465e270342
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
apiserver/bll/task/task_operations.py
View File

@ -367,7 +367,21 @@ def stop_task(
and (datetime.utcnow() - t.last_update).total_seconds() < update_timeout and (datetime.utcnow() - t.last_update).total_seconds() < update_timeout
) )
if TaskSystemTags.development in task.system_tags or not is_run_by_worker(task): is_queued = task.status == TaskStatus.queued
set_stopped = (
is_queued
or TaskSystemTags.development in task.system_tags
or not is_run_by_worker(task)
)
if set_stopped:
if is_queued:
try:
TaskBLL.dequeue(task, company_id=company_id, silent_fail=True)
except APIError:
# dequeue may fail if the task was not enqueued
pass
new_status = TaskStatus.stopped new_status = TaskStatus.stopped
status_message = f"Stopped by {user_name}" status_message = f"Stopped by {user_name}"
else: else: