From 28b85028fe4da3ab963b69e8ac0f7feef73cfcf6 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Thu, 4 Feb 2021 10:41:51 +0200
Subject: [PATCH] Fix close task after logger is closed

---
 clearml/task.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/clearml/task.py b/clearml/task.py
index f63a0d23..0644ef5e 100644
--- a/clearml/task.py
+++ b/clearml/task.py
@@ -2954,6 +2954,11 @@ class Task(_Task):
                     self._resource_monitor.stop()
                     self._resource_monitor = None
 
+                if self._logger:
+                    self._logger.set_flush_period(None)
+                    # noinspection PyProtectedMember
+                    self._logger._close_stdout_handler(wait=wait_for_uploads or wait_for_std_log)
+
                 if not is_sub_process:
                     # change task status
                     if not task_status:
@@ -2965,11 +2970,6 @@ class Task(_Task):
                     elif task_status[0] == 'stopped':
                         self.stopped()
 
-                if self._logger:
-                    self._logger.set_flush_period(None)
-                    # noinspection PyProtectedMember
-                    self._logger._close_stdout_handler(wait=wait_for_uploads or wait_for_std_log)
-
                 # this is so in theory we can close a main task and start a new one
                 if self.is_main_task():
                     Task.__main_task = None