We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 68885dc commit 5a62b7bCopy full SHA for 5a62b7b
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,7 @@ 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
+ s3_runproc_sh = self._create_and_upload_runproc(script, self.output_kms_key, entrypoint_s3_uri)
1809
1810
return s3_runproc_sh, inputs, job_name
1811
0 commit comments