Documentation of @step decorator

[JurijGass]

Hi, I'm trying to implement a pipeline with @step decorator. So I was following descriptions on this side:
https://docs.aws.amazon.com/sagemaker/latest/dg/pipelines-step-decorator-create-pipeline.html

Paragraph: Define custom dependencies
Code Line: get_step(step_train_result).add_depends_on(step_process_result)
Problem: Function 'add_depends_on' needs a list.
Correct version would be: get_step(step_train_result).add_depends_on([step_process_result])

No description of the function 'add_depends_on' in sdk docs: https://sagemaker.readthedocs.io/en/stable/workflows/pipelines/sagemaker.workflow.pipelines.html#sagemaker.workflow.steps.Step
Found definition in the code:

sagemaker-python-sdk/src/sagemaker/workflow/steps.py

Line 158 in 87d661a

def add_depends_on(self, step_names: List[Union[str, "Step", "StepCollection", StepOutput]]):

[nmadan]

Hi @JurijGass, thanks for your feedback. We are currently working on getting this fixed!

[nmadan]

Documentation has been fixed: https://docs.aws.amazon.com/sagemaker/latest/dg/pipelines-step-decorator-create-pipeline.html#pipelines-step-decorator-run-pipeline-link-custom