From de6ea269f89bfa055095f81555e587242350221d Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Wed, 5 Jan 2022 08:56:56 +0200
Subject: [PATCH] Fix Pipeline controller function support for dict as input
 argument

---
 clearml/automation/controller.py | 4 ++--
 clearml/automation/trigger.py    | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/clearml/automation/controller.py b/clearml/automation/controller.py
index 98d31cbf..d4ac4631 100644
--- a/clearml/automation/controller.py
+++ b/clearml/automation/controller.py
@@ -1143,10 +1143,10 @@ class PipelineController(object):
                 self._task._set_configuration(
                     name=self._config_section, config_type='dictionary',
                     config_text=json.dumps(pipeline_dag, indent=2))
-                params.update(self._pipeline_args)
+                params.update(flatten_dictionary(self._pipeline_args))
                 # noinspection PyProtectedMember
                 self._task._set_parameters(
-                    {'{}/{}'.format(self._args_section, k): str(v) for k, v in params.items()},
+                    {'{}/{}'.format(self._args_section, k): v for k, v in params.items()},
                     __parameters_descriptions=self._pipeline_args_desc,
                     __update=True,
                 )
diff --git a/clearml/automation/trigger.py b/clearml/automation/trigger.py
index aa2f5757..3b01bff8 100644
--- a/clearml/automation/trigger.py
+++ b/clearml/automation/trigger.py
@@ -247,7 +247,7 @@ class TriggerScheduler(BaseScheduler):
         (skip until the next scheduled time period). Default False.
         :param reuse_task: If True, re-enqueue the same Task (i.e. do not clone it) every time, default False.
         :param task_parameters: Configuration parameters to the executed Task.
-        for example: {'Args/batch': '12'} Notice: not available when reuse_task=True/
+        for example: {'Args/batch': '12'} Notice: not available when reuse_task=True
         :param task_overrides: Change task definition.
         for example {'script.version_num': None, 'script.branch': 'main'} Notice: not available when reuse_task=True
         :return: True if job is successfully added to the scheduling list