We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents f04d38c + 5a5b3d8 commit 739b2daCopy full SHA for 739b2da
src/sagemaker/processing.py
@@ -1566,6 +1566,7 @@ def _create_estimator(
1566
sagemaker_session=self.sagemaker_session,
1567
debugger_hook_config=False,
1568
disable_profiler=True,
1569
+ output_kms_key=self.output_kms_key,
1570
)
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.`")
1806
1807
script = estimator.uploaded_code.script_name
- 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
+ )
1812
1813
return s3_runproc_sh, inputs, job_name
1814
0 commit comments