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'}
- 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.

View File

@ -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)))