Skip to content

Commit 0d35b2a

Browse files
Simon Zamarinvikas0203
Simon Zamarin
authored andcommitted
fixed issue with FrameworkProcessor not using the specified output_kms_key when uploading sourcedir.tar.gz
1 parent 5d59767 commit 0d35b2a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sagemaker/processing.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1493,6 +1493,7 @@ def _create_estimator(
14931493
enable_network_isolation=False, # True -> uploads to input channel. Not what we want!
14941494
image_uri=self.image_uri,
14951495
role=self.role,
1496+
output_kms_key=self.output_kms_key,
14961497
# Estimator instance_count doesn't currently matter to FrameworkProcessor, and the
14971498
# SKLearn Framework Estimator requires instance_type==1. So here we hard-wire it to 1,
14981499
# but if it matters in future perhaps we could take self.instance_count here and have
@@ -1801,7 +1802,7 @@ def _upload_payload(
18011802
entry_point=entry_point,
18021803
source_dir=source_dir,
18031804
dependencies=dependencies,
1804-
git_config=git_config,
1805+
git_config=git_config
18051806
)
18061807

18071808
estimator._prepare_for_training(job_name=job_name)

0 commit comments

Comments
 (0)