File tree 1 file changed +4
-2
lines changed
tests/integ/sagemaker/workflow
1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -566,7 +566,7 @@ def test_multi_step_framework_processing_pipeline_different_source_dir(
566
566
execution .wait (delay = 540 , max_attempts = 3 )
567
567
except WaiterError :
568
568
pass
569
-
569
+
570
570
execution_steps = execution .list_steps ()
571
571
assert len (execution_steps ) == 2
572
572
for step in execution_steps :
@@ -1034,6 +1034,8 @@ def _verify_code_artifacts_of_framework_processing_step(
1034
1034
assert expected_step_artifact in s3_code_objects
1035
1035
1036
1036
# verify runprocs contain the correct commands
1037
- step_runproc = S3Downloader .read_file (f"s3://{ bucket } /{ expected_step_artifact } " )
1037
+ step_runproc = S3Downloader .read_file (
1038
+ f"s3://{ bucket } /{ expected_step_artifact } " , pipeline_session
1039
+ )
1038
1040
assert f"python { entry_point } " in step_runproc
1039
1041
return source_dir , expected_step_artifact
You can’t perform that action at this time.
0 commit comments