Version bump to v1.9.2

apiserver/bll/task/task_bll.py

@@ -256,6 +256,16 @@ class TaskBLL:
                 not in [TaskSystemTags.development, EntityVisibility.archived.value]
             ]
 
+        def ensure_int_labels(execution: dict) -> dict:
+            if not execution:
+                return execution
+
+            model_labels = execution.get("model_labels")
+            if model_labels:
+                execution["model_labels"] = {k: int(v) for k, v in model_labels.items()}
+
+            return execution
+
         parent_task = (
             task.parent
             if task.parent and not task.parent.startswith(deleted_prefix)
@@ -280,7 +290,7 @@ class TaskBLL:
             output=Output(destination=task.output.destination) if task.output else None,
             models=Models(input=input_models or task.models.input),
             container=escape_dict(container) or task.container,
-            execution=execution_dict,
+            execution=ensure_int_labels(execution_dict),
             configuration=params_dict.get("configuration") or task.configuration,
             hyperparams=params_dict.get("hyperparams") or task.hyperparams,
         )

apiserver/services/reports.py

@@ -80,7 +80,7 @@ def update_report(call: APICall, company_id: str, request: UpdateReportRequest):
     if not partial_update_dict:
         return UpdateResponse(updated=0)
 
-    allowed_for_published = set(partial_update_dict.keys()).issubset({"tags", "name"})
+    allowed_for_published = set(partial_update_dict.keys()).issubset({"tags", "name", "comment"})
     if task.status != TaskStatus.created and not allowed_for_published:
         raise errors.bad_request.InvalidTaskStatus(
             expected=TaskStatus.created, status=task.status

apiserver/tests/automated/test_reports.py

@@ -56,13 +56,16 @@ class TestReports(TestService):
         self.assertEqual(ret.tags, [])
         self.api.reports.publish(task=task_id)
         with self.api.raises(errors.bad_request.InvalidTaskStatus):
-            self.api.reports.update(task=task_id, comment=comment)
+            self.api.reports.update(task=task_id, report="New report text")
 
         # update on tags or rename can be done for published report too
-        self.api.reports.update(task=task_id, name="new name", tags=["test"])
+        self.api.reports.update(
+            task=task_id, name="new name", tags=["test"], comment="Yet another comment"
+        )
         task = self.api.tasks.get_all_ex(id=[task_id]).tasks[0]
         self.assertEqual(task.tags, ["test"])
         self.assertEqual(task.name, "new name")
+        self.assertEqual(task.comment, "Yet another comment")
 
         # move under another project autodeletes the empty project
         new_project_name = "Reports Test"

apiserver/tests/automated/test_tasks_edit.py

@@ -103,7 +103,7 @@ class TestTasksEdit(TestService):
 
         new_name = "new test"
         new_tags = ["by"]
-        execution_overrides = dict(framework="Caffe")
+        execution_overrides = dict(framework="Caffe", model_labels={"test": 1.0})
         new_task_id = self._clone_task(
             task=task,
             new_task_name=new_name,
@@ -120,6 +120,7 @@ class TestTasksEdit(TestService):
         self.assertEqual(new_task.parent, task)
         # self.assertEqual(new_task.execution.parameters, execution["parameters"])
         self.assertEqual(new_task.execution.framework, execution_overrides["framework"])
+        self.assertEqual(new_task.execution.model_labels, {"test": 1})
         self.assertEqual(new_task.system_tags, ["test"])
 
     def test_model_check_in_clone(self):

apiserver/version.py

@@ -1 +1 @@
-__version__ = "1.9.1"
+__version__ = "1.9.2"

docker/docker-compose.yml

@@ -172,7 +172,7 @@ services:
       CLEARML_API_SECRET_KEY: ${CLEARML_API_SECRET_KEY:-}
       CLEARML_AGENT_GIT_USER: ${CLEARML_AGENT_GIT_USER}
       CLEARML_AGENT_GIT_PASS: ${CLEARML_AGENT_GIT_PASS}
-      CLEARML_AGENT_UPDATE_VERSION: ${CLEARML_AGENT_UPDATE_VERSION:-">=0.17.0"}
+      CLEARML_AGENT_UPDATE_VERSION: ${CLEARML_AGENT_UPDATE_VERSION:->=0.17.0}
       CLEARML_AGENT_DEFAULT_BASE_DOCKER: "ubuntu:18.04"
       AWS_ACCESS_KEY_ID: ${AWS_ACCESS_KEY_ID:-}
       AWS_SECRET_ACCESS_KEY: ${AWS_SECRET_ACCESS_KEY:-}