Skip to content

Commit a7d3d63

Browse files
committed
feature: support displayName and description for pipeline steps
1 parent 80d5fb7 commit a7d3d63

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/sagemaker/workflow/lambda_step.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ def __init__(
105105
depends_on (List[str]): A list of step names this `sagemaker.workflow.steps.LambdaStep`
106106
depends on
107107
"""
108-
super(LambdaStep, self).__init__(name,display_name, description, StepTypeEnum.LAMBDA, depends_on)
108+
super(LambdaStep, self).__init__(name, display_name, description, StepTypeEnum.LAMBDA, depends_on)
109109
self.lambda_func = lambda_func
110110
self.outputs = outputs if outputs is not None else []
111111
self.cache_config = cache_config

tests/unit/sagemaker/workflow/test_lambda_step.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,8 @@ def test_lambda_step(sagemaker_session):
4848
function_arn="arn:aws:lambda:us-west-2:123456789012:function:sagemaker_test_lambda",
4949
session=sagemaker_session,
5050
),
51-
display_name="MyLambdaStep", description="MyLambdaStepDescription",
51+
display_name="MyLambdaStep",
52+
description="MyLambdaStepDescription",
5253
inputs={"arg1": "foo", "arg2": 5, "arg3": param},
5354
outputs=[outputParam1, outputParam2],
5455
)

0 commit comments

Comments
 (0)