Skip to content

Commit 976a2d7

Browse files
author
Brock Wade
committed
include pipeline session in test pipeline objects
1 parent 547c895 commit 976a2d7

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/integ/sagemaker/workflow/test_processing_steps.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -435,7 +435,9 @@ def test_multi_step_framework_processing_pipeline_same_source_dir(
435435
cache_config=cache_config,
436436
)
437437

438-
pipeline = Pipeline(name=pipeline_name, steps=[step_1, step_2])
438+
pipeline = Pipeline(
439+
name=pipeline_name, steps=[step_1, step_2], sagemaker_session=pipeline_session
440+
)
439441
try:
440442
pipeline.create(role)
441443
definition = json.loads(pipeline.definition())
@@ -526,7 +528,9 @@ def test_multi_step_framework_processing_pipeline_different_source_dir(
526528
cache_config=cache_config,
527529
)
528530

529-
pipeline = Pipeline(name=pipeline_name, steps=[step_1, step_2])
531+
pipeline = Pipeline(
532+
name=pipeline_name, steps=[step_1, step_2], sagemaker_session=pipeline_session
533+
)
530534
try:
531535
pipeline.create(role)
532536
definition = json.loads(pipeline.definition())

0 commit comments

Comments
 (0)