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(
1566
1566
sagemaker_session = self .sagemaker_session ,
1567
1567
debugger_hook_config = False ,
1568
1568
disable_profiler = True ,
1569
+ output_kms_key = self .output_kms_key ,
1569
1570
)
1570
1571
1571
1572
def get_run_args (
@@ -1804,7 +1805,10 @@ def _pack_and_upload_code(
1804
1805
raise RuntimeError ("S3 source_dir file must be named `sourcedir.tar.gz.`" )
1805
1806
1806
1807
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
+ )
1808
1812
1809
1813
return s3_runproc_sh , inputs , job_name
1810
1814
You can’t perform that action at this time.
0 commit comments