Fix Pipeline Controller automagic framework connect

This commit is contained in:
allegroai 2022-03-27 14:02:34 +03:00
parent ce7e77a00e
commit 18bf4ed04d
2 changed files with 8 additions and 0 deletions

View File

@ -295,6 +295,8 @@ class PipelineController(object):
'container.arguments': '--ipc=host'} 'container.arguments': '--ipc=host'}
- match container image to a previous step - match container image to a previous step
task_overrides={'container.image': '${stage1.container.image}'} 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. :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 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. :param monitor_metrics: Optional, log the step's metrics on the pipeline Task.

View File

@ -589,6 +589,12 @@ if __name__ == '__main__':
:param _sanitize_helper_functions: Sanitization function for the helper function string. :param _sanitize_helper_functions: Sanitization function for the helper function string.
:return: Newly created Task object :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_frameworks or isinstance(auto_connect_frameworks, (bool, dict)))
assert (not auto_connect_arg_parser or isinstance(auto_connect_arg_parser, (bool, dict))) assert (not auto_connect_arg_parser or isinstance(auto_connect_arg_parser, (bool, dict)))