Skip to content

Commit 86e5dae

Browse files
committed
fix: Pass KMS value to create processing job in S3 bucket with restrictions
1 parent ace07d7 commit 86e5dae

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/sagemaker/processing.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1447,6 +1447,7 @@ def _create_estimator(
14471447
sagemaker_session=self.sagemaker_session,
14481448
debugger_hook_config=False,
14491449
disable_profiler=True,
1450+
output_kms_key=self.output_kms_key,
14501451
)
14511452

14521453
def get_run_args(
@@ -1677,6 +1678,7 @@ def _pack_and_upload_code(self, code, source_dir, dependencies, git_config, job_
16771678
self._generate_framework_script(script),
16781679
desired_s3_uri=entrypoint_s3_uri,
16791680
sagemaker_session=self.sagemaker_session,
1681+
kms_key=self.output_kms_key,
16801682
)
16811683
logger.info("runproc.sh uploaded to %s", s3_runproc_sh)
16821684

0 commit comments

Comments
 (0)