Skip to content

Commit 5ff27c0

Browse files
author
Rohan Gujarathi
committed
fix: another unit tests fix
1 parent 9797022 commit 5ff27c0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/unit/sagemaker/workflow/test_lambda_step.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,8 @@ def test_pipeline_interpolates_lambda_outputs(sagemaker_session):
106106
name="MyLambdaStep2",
107107
depends_on=["TestStep"],
108108
lambda_func=Lambda(
109-
function_arn="arn:aws:lambda:us-west-2:123456789012:function:sagemaker_test_lambda"
109+
function_arn="arn:aws:lambda:us-west-2:123456789012:function:sagemaker_test_lambda",
110+
session=sagemaker_session,
110111
),
111112
inputs={"arg1": outputParam1},
112113
outputs=[outputParam2],

0 commit comments

Comments
 (0)