ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-12 07:35:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add Pipeline start_controller_locally implementation in decorator

Browse Source
This commit is contained in:
allegroai 2022-09-02 23:38:43 +03:00
parent a663805eb7
commit 1f63946c3f
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/automation/controller.py
View File

@ -3221,6 +3221,9 @@ class PipelineDecorator(PipelineController):
- paramB: sectionB/paramB - paramB: sectionB/paramB
- paramC: sectionB/paramC - paramC: sectionB/paramC
- paramD: Args/paramD - paramD: Args/paramD
:param start_controller_locally: If True, start the controller on the local machine. The steps will run
remotely if `PipelineDecorator.run_locally` or `PipelineDecorator.debug_pipeline` are not called.
Default: False
""" """
def decorator_wrap(func): def decorator_wrap(func):
@ -3331,7 +3334,8 @@ class PipelineDecorator(PipelineController):
pipeline_kwargs[k] = a_pipeline.get_parameters()[k] pipeline_kwargs[k] = a_pipeline.get_parameters()[k]
# run the actual pipeline # run the actual pipeline
if not PipelineDecorator._debug_execute_step_process and pipeline_execution_queue: if not start_controller_locally and \
not PipelineDecorator._debug_execute_step_process and pipeline_execution_queue:
# rerun the pipeline on a remote machine # rerun the pipeline on a remote machine
a_pipeline._task.execute_remotely(queue_name=pipeline_execution_queue) a_pipeline._task.execute_remotely(queue_name=pipeline_execution_queue)
# when we get here it means we are running remotely # when we get here it means we are running remotely