From a843ef3321beea656186b0d74a0e27cfac4e102d Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Mon, 9 Oct 2023 13:51:38 +0300
Subject: [PATCH] Make sure we delete old tasks artifcats as well when deleting
 old session tasks

---
 clearml_session/__main__.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/clearml_session/__main__.py b/clearml_session/__main__.py
index 79a0d45..0c83715 100644
--- a/clearml_session/__main__.py
+++ b/clearml_session/__main__.py
@@ -301,9 +301,13 @@ def delete_old_tasks(state, client, base_task_id):
         if state.get('verbose'):
             print('Removing {}/{} stale sessions'.format(i+1, len(previous_tasks)))
         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:
             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):