File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff 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,10 @@ 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+ evaluated_kms_key = kms_key if kms_key else self .output_kms_key
1809+ s3_runproc_sh = self ._create_and_upload_runproc (
1810+ script , evaluated_kms_key , entrypoint_s3_uri
1811+ )
18081812
18091813 return s3_runproc_sh , inputs , job_name
18101814
You can’t perform that action at this time.
0 commit comments