From 5a62b7bb8dc74631dae65c0ed1298ef99123ad76 Mon Sep 17 00:00:00 2001 From: Raj Oli Date: Mon, 19 Sep 2022 12:03:00 +1000 Subject: [PATCH] fix: Pass KMS value to create processing job in S3 bucket with restrictions --- src/sagemaker/processing.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/sagemaker/processing.py b/src/sagemaker/processing.py index 228f5068a1..12acd55923 100644 --- a/src/sagemaker/processing.py +++ b/src/sagemaker/processing.py @@ -1566,6 +1566,7 @@ def _create_estimator( sagemaker_session=self.sagemaker_session, debugger_hook_config=False, disable_profiler=True, + output_kms_key=self.output_kms_key, ) def get_run_args( @@ -1804,7 +1805,7 @@ def _pack_and_upload_code( raise RuntimeError("S3 source_dir file must be named `sourcedir.tar.gz.`") script = estimator.uploaded_code.script_name - s3_runproc_sh = self._create_and_upload_runproc(script, kms_key, entrypoint_s3_uri) + s3_runproc_sh = self._create_and_upload_runproc(script, self.output_kms_key, entrypoint_s3_uri) return s3_runproc_sh, inputs, job_name