From 18bf4ed04d4c25b9349fb969568ce50618ef44c0 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sun, 27 Mar 2022 14:02:34 +0300
Subject: [PATCH] Fix Pipeline Controller automagic framework connect

---
 clearml/automation/controller.py           | 2 ++
 clearml/backend_interface/task/populate.py | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/clearml/automation/controller.py b/clearml/automation/controller.py
index 4af50821..11155b55 100644
--- a/clearml/automation/controller.py
+++ b/clearml/automation/controller.py
@@ -295,6 +295,8 @@ class PipelineController(object):
                                 'container.arguments': '--ipc=host'}
             - match container image to a previous step
                 task_overrides={'container.image': '${stage1.container.image}'}
+            - reset requirements (the agent will use the "requirements.txt" inside the repo)
+                task_overrides={'script.requirements.pip': ""}
         :param execution_queue: Optional, the queue to use for executing this specific step.
             If not provided, the task will be sent to the default execution queue, as defined on the class
         :param monitor_metrics: Optional, log the step's metrics on the pipeline Task.
diff --git a/clearml/backend_interface/task/populate.py b/clearml/backend_interface/task/populate.py
index 6ff92f29..d94baaef 100644
--- a/clearml/backend_interface/task/populate.py
+++ b/clearml/backend_interface/task/populate.py
@@ -589,6 +589,12 @@ if __name__ == '__main__':
         :param _sanitize_helper_functions: Sanitization function for the helper function string.
         :return: Newly created Task object
         """
+        # not set -> equals True
+        if auto_connect_frameworks is None:
+            auto_connect_frameworks = True
+        if auto_connect_arg_parser is None:
+            auto_connect_arg_parser = True
+
         assert (not auto_connect_frameworks or isinstance(auto_connect_frameworks, (bool, dict)))
         assert (not auto_connect_arg_parser or isinstance(auto_connect_arg_parser, (bool, dict)))