Skip to content

Commit 739b2da

Browse files
authored
Merge branch 'master' into feat/jumpstart-model-estimator-classes
2 parents f04d38c + 5a5b3d8 commit 739b2da

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/sagemaker/processing.py

+5-1
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)