fix: Pass KMS value to create processing job in S3 bucket with restrictions

raj-oli · raj-oli · commit ca444ffd2b4f · 2022-10-26T10:54:52.000+11:00
diff --git a/src/sagemaker/processing.py b/src/sagemaker/processing.py
@@ -1456,6 +1456,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(
@@ -1689,6 +1690,7 @@ def _pack_and_upload_code(
             desired_s3_uri=entrypoint_s3_uri,
             kms_key=kms_key,
             sagemaker_session=self.sagemaker_session,
+            kms_key=self.output_kms_key,
         )
         logger.info("runproc.sh uploaded to %s", s3_runproc_sh)
 

Original file line number	Diff line number	Diff line change
`@@ -1456,6 +1456,7 @@ def _create_estimator(`
`1456`	`1456`	`sagemaker_session=self.sagemaker_session,`
`1457`	`1457`	`debugger_hook_config=False,`
`1458`	`1458`	`disable_profiler=True,`
	`1459`	`+ output_kms_key=self.output_kms_key,`
`1459`	`1460`	`)`
`1460`	`1461`
`1461`	`1462`	`def get_run_args(`
`@@ -1689,6 +1690,7 @@ def _pack_and_upload_code(`
`1689`	`1690`	`desired_s3_uri=entrypoint_s3_uri,`
`1690`	`1691`	`kms_key=kms_key,`
`1691`	`1692`	`sagemaker_session=self.sagemaker_session,`
	`1693`	`+ kms_key=self.output_kms_key,`
`1692`	`1694`	`)`
`1693`	`1695`	`logger.info("runproc.sh uploaded to %s", s3_runproc_sh)`
`1694`	`1696`