ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 17:17:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add support for references in Pipeline queue_name (#857)

Browse Source
This commit is contained in:
Anton Cherepkov 2023-01-26 21:11:47 +04:00 committed by GitHub
parent da6f75363d
commit 54c601eea2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
clearml/automation/controller.py
View File

@ -1925,7 +1925,8 @@ class PipelineController(object):
node.job.task.get_logger().report_text(
"\nNode '{}' failed. Retrying... (this is retry number {})\n".format(node.name, self._retries[node.name])
)
node.job.launch(queue_name=node.queue or self._default_execution_queue)
parsed_queue_name = self._parse_step_ref(node.queue)
node.job.launch(queue_name=parsed_queue_name or self._default_execution_queue)
def _launch_node(self, node):
# type: (PipelineController.Node) -> ()
@ -2014,7 +2015,9 @@ class PipelineController(object):
self._running_nodes.append(node.name)
else:
self._running_nodes.append(node.name)
return node.job.launch(queue_name=node.queue or self._default_execution_queue)
parsed_queue_name = self._parse_step_ref(node.queue)
return node.job.launch(queue_name=parsed_queue_name or self._default_execution_queue)
return True