From 4ca9dc4c00fe972786db15a01ce718c051b28ed6 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Thu, 15 Oct 2020 23:18:20 +0300
Subject: [PATCH] Add Task.set_task_type()

---
 trains/backend_interface/task/task.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/trains/backend_interface/task/task.py b/trains/backend_interface/task/task.py
index 1ce94890..7cb12a1b 100644
--- a/trains/backend_interface/task/task.py
+++ b/trains/backend_interface/task/task.py
@@ -1190,6 +1190,20 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
         self._set_task_property("comment", str(comment))
         self._edit(comment=comment)
 
+    def set_task_type(self, task_type):
+        # type: (Union[str, TaskTypes]) -> ()
+        """
+        Set the task_type for the Task.
+
+        :param task_type: The task_type of the Task (see optional values in TaskTypes).
+        :type task_type: str or TaskTypes
+        """
+        if not isinstance(task_type, self.TaskTypes):
+            task_type = self.TaskTypes(task_type)
+
+        self._set_task_property("task_type", str(task_type))
+        self._edit(type=task_type)
+
     def set_initial_iteration(self, offset=0):
         # type: (int) -> int
         """