Skip to content

Commit 7732b9e

Browse files
author
Brock Wade
committed
fix: add pipeline session s3 downloader
1 parent 89e66df commit 7732b9e

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/integ/sagemaker/workflow/test_processing_steps.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -566,7 +566,7 @@ def test_multi_step_framework_processing_pipeline_different_source_dir(
566566
execution.wait(delay=540, max_attempts=3)
567567
except WaiterError:
568568
pass
569-
569+
570570
execution_steps = execution.list_steps()
571571
assert len(execution_steps) == 2
572572
for step in execution_steps:
@@ -1034,6 +1034,8 @@ def _verify_code_artifacts_of_framework_processing_step(
10341034
assert expected_step_artifact in s3_code_objects
10351035

10361036
# 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+
)
10381040
assert f"python {entry_point}" in step_runproc
10391041
return source_dir, expected_step_artifact

0 commit comments

Comments
 (0)