ClearML/clearml-session
1
0
Fork 0
mirror of https://github.com/clearml/clearml-session synced 2025-02-07 05:20:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make sure we delete old tasks artifcats as well when deleting old session tasks

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

6
clearml_session/__main__.py
View File

@ -301,9 +301,13 @@ def delete_old_tasks(state, client, base_task_id):
if state.get('verbose'): if state.get('verbose'):
print('Removing {}/{} stale sessions'.format(i+1, len(previous_tasks))) print('Removing {}/{} stale sessions'.format(i+1, len(previous_tasks)))
try: try:
client.tasks.delete(task=t.id, force=True) Task.get_task(task_id=t.id).delete(delete_artifacts_and_models=True, skip_models_used_by_other_tasks=True, raise_on_error=True)
except Exception as ex: except Exception as ex:
logging.getLogger().warning('{}\nFailed deleting old session {}'.format(ex, t.id)) logging.getLogger().warning('{}\nFailed deleting old session {}'.format(ex, t.id))
try:
client.tasks.delete(task=t.id, force=True)
except Exception as ex:
logging.getLogger().warning('{}\nFailed deleting old session {}'.format(ex, t.id))
def _get_running_tasks(client, prev_task_id): def _get_running_tasks(client, prev_task_id):