Skip to content

Commit 5a62b7b

Browse files
committed
fix: Pass KMS value to create processing job in S3 bucket with restrictions
1 parent 68885dc commit 5a62b7b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sagemaker/processing.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1566,6 +1566,7 @@ def _create_estimator(
15661566
sagemaker_session=self.sagemaker_session,
15671567
debugger_hook_config=False,
15681568
disable_profiler=True,
1569+
output_kms_key=self.output_kms_key,
15691570
)
15701571

15711572
def get_run_args(
@@ -1804,7 +1805,7 @@ def _pack_and_upload_code(
18041805
raise RuntimeError("S3 source_dir file must be named `sourcedir.tar.gz.`")
18051806

18061807
script = estimator.uploaded_code.script_name
1807-
s3_runproc_sh = self._create_and_upload_runproc(script, kms_key, entrypoint_s3_uri)
1808+
s3_runproc_sh = self._create_and_upload_runproc(script, self.output_kms_key, entrypoint_s3_uri)
18081809

18091810
return s3_runproc_sh, inputs, job_name
18101811

0 commit comments

Comments
 (0)