From 32941467615b92b1c32a9a27ce4d6b894cd79f2f Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Wed, 5 Jul 2023 08:41:35 +0300
Subject: [PATCH] Future proof clearml package support

---
 clearml_session/__main__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/clearml_session/__main__.py b/clearml_session/__main__.py
index cc512d9..65735d6 100644
--- a/clearml_session/__main__.py
+++ b/clearml_session/__main__.py
@@ -589,7 +589,7 @@ def wait_for_machine(state, task):
             if stopped_counter > 5:
                 break
         # noinspection PyProtectedMember
-        status, message = task._get_status()
+        status, message = task.get_status_message() if hasattr(task, "get_status_message") else task._get_status()
         status = str(status)
         if last_status != status or last_message != message:
             # noinspection PyProtectedMember