From 7ce5bc031353dacb427f33492c039e41184275a9 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Thu, 15 Oct 2020 23:19:05 +0300
Subject: [PATCH] Fix wrong argument order (issue #220)

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

diff --git a/trains/backend_interface/task/task.py b/trains/backend_interface/task/task.py
index 7cb12a1b..ec1d300d 100644
--- a/trains/backend_interface/task/task.py
+++ b/trains/backend_interface/task/task.py
@@ -657,7 +657,8 @@ class Task(IdObjectBase, AccessMixin, SetupUploadMixin):
         :type tags: [str]
         """
         self._conditionally_start_task()
-        self._get_output_model(upload_required=False).update_for_task(model_uri, self.id, name, comment, tags)
+        self._get_output_model(upload_required=False).update_for_task(
+            uri=model_uri, task_id=self.id, name=name, comment=comment, tags=tags)
 
     def update_output_model_and_upload(
             self,