From 88e4f8db82dd4f2dfa980d198605e791a1eed81a Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sat, 16 Oct 2021 23:11:25 +0300
Subject: [PATCH] Add Task.mark_completed() force and status_message args

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

diff --git a/clearml/backend_interface/task/task.py b/clearml/backend_interface/task/task.py
index a1cb4893..dfb3f03b 100644
--- a/clearml/backend_interface/task/task.py
+++ b/clearml/backend_interface/task/task.py
@@ -566,12 +566,18 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
         warnings.warn("'completed' is deprecated; use 'mark_completed' instead.", DeprecationWarning)
         return self.mark_completed(ignore_errors=ignore_errors)
 
-    def mark_completed(self, ignore_errors=True):
-        # type: (bool) -> ()
+    def mark_completed(self, ignore_errors=True, status_message=None, force=False):
+        # type: (bool, Optional[str], bool) -> ()
         """ The signal indicating that this Task completed. """
         if hasattr(tasks, 'CompletedRequest') and callable(tasks.CompletedRequest):
-            return self.send(tasks.CompletedRequest(self.id, status_reason='completed'), ignore_errors=ignore_errors)
-        return self.send(tasks.StoppedRequest(self.id, status_reason='completed'), ignore_errors=ignore_errors)
+            return self.send(
+                tasks.CompletedRequest(self.id, status_reason='completed', status_message=status_message, force=force),
+                ignore_errors=ignore_errors
+            )
+        return self.send(
+            tasks.StoppedRequest(self.id, status_reason='completed', status_message=status_message, force=force),
+            ignore_errors=ignore_errors
+        )
 
     def mark_failed(self, ignore_errors=True, status_reason=None, status_message=None, force=False):
         # type: (bool, Optional[str], Optional[str], bool) -> ()