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