ClearML/clearml-server
1
0
Fork 0
mirror of https://github.com/clearml/clearml-server synced 2025-05-13 08:11:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix task cleanup

Browse Source
This commit is contained in:
allegroai 2022-07-08 17:40:55 +03:00
parent 05357fe25e
commit 6683d2d7a9
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
apiserver/bll/task/task_cleanup.py
View File

@ -181,25 +181,25 @@ def cleanup_task(
model_urls = {m.uri for m in models.draft.objects().only("uri") if m.uri} model_urls = {m.uri for m in models.draft.objects().only("uri") if m.uri}
deleted_task_id = f"{deleted_prefix}{task.id}" deleted_task_id = f"{deleted_prefix}{task.id}"
updated_children = 0
if update_children: if update_children:
with TimingContext("mongo", "update_task_children"): with TimingContext("mongo", "update_task_children"):
updated_children = Task.objects(parent=task.id).update( updated_children = Task.objects(parent=task.id).update(
parent=deleted_task_id parent=deleted_task_id
) )
else:
updated_children = 0
if models.draft and delete_output_models: deleted_models = 0
with TimingContext("mongo", "delete_models"): updated_models = 0
if models.draft:
if delete_output_models:
deleted_models = models.draft.objects().delete() deleted_models = models.draft.objects().delete()
else: elif update_children:
deleted_models = 0 updated_models = models.draft.objects().update(task=deleted_task_id)
else:
models.draft.objects().update(unset__task=1)
if models.published and update_children: if models.published and update_children:
with TimingContext("mongo", "update_task_models"): updated_models += models.published.objects().update(task=deleted_task_id)
updated_models = models.published.objects().update(task=deleted_task_id)
else:
updated_models = 0
event_bll.delete_task_events(task.company, task.id, allow_locked=force) event_bll.delete_task_events(task.company, task.id, allow_locked=force)