Azure / Azure/MachineLearningNotebooks
StepRun state management in Pipeline runtime
- 主要语言
- Jupyter Notebook
- 星标
- 4.4k
- 派生
- 2.6k
- PR 合并指标
- 30 天内没有已合并 PR
描述
Consider python azureml-sdk Pipeline with three PythonScriptStep [step_1, step_a, step_b] and edges:
step_a.run_after(step_1)
step_b.run_after(step_1)
where in step_1 runtime "a condition" is checked based on which we want to cancel running step_b but want to continue running pipeline (ie. step_b). Please find a graph appended.

I tried to implement it in two ways:
_from azureml.core import Run
from azureml.pipeline.core.run import PipelineRun_
1) Deleting node in graph (Is this API executed on instance of Graph object for only this Pipeline run?)
_if a_condition:
run = Run.get_context()
pipeline_run = PipelineRun(run.experiment, run.id)
graph=pipeline_run.get_graph()
for n in graph.nodes:
if n.name =='step_1':
graph.delete_node(n.node_id)_
Nonetheless:
azureml/pipeline/core/graph.py in delete_node(self, node_id)
raise NotImplementedError
2) Having some higher level API in the picture of:
_from azureml.core import Run
from azureml.pipeline.core.run import PipelineRun
if a_condition:
run = Run.get_context()
pipeline_run = PipelineRun(run.experiment, run.id)
step_a_run=pipeline_run.find_step_run(step_a_name)
step_a_run.cancel() #alternatively .fail()_
_step_a_run_ is an empty list as _step_a_ have not started to run. Besides _StepRun_ inherits _cancel()_ after _Run_ so I it cancels entire pipeline.
Is there some work planned in Azure ML Python SDK for Pipeline(or its Graph) steps management during its runtime? Especially this scenario to cancel running selected before it starts to run/prepare?
Its useful when you have highly automated pipelines.
贡献指南
这个仓库没有索引到贡献指南
评估
这个 Issue 还没有评估数据。