File tree 1 file changed +2
-1
lines changed
1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -1493,6 +1493,7 @@ def _create_estimator(
1493
1493
enable_network_isolation = False , # True -> uploads to input channel. Not what we want!
1494
1494
image_uri = self .image_uri ,
1495
1495
role = self .role ,
1496
+ output_kms_key = self .output_kms_key ,
1496
1497
# Estimator instance_count doesn't currently matter to FrameworkProcessor, and the
1497
1498
# SKLearn Framework Estimator requires instance_type==1. So here we hard-wire it to 1,
1498
1499
# but if it matters in future perhaps we could take self.instance_count here and have
@@ -1801,7 +1802,7 @@ def _upload_payload(
1801
1802
entry_point = entry_point ,
1802
1803
source_dir = source_dir ,
1803
1804
dependencies = dependencies ,
1804
- git_config = git_config ,
1805
+ git_config = git_config
1805
1806
)
1806
1807
1807
1808
estimator ._prepare_for_training (job_name = job_name )
You can’t perform that action at this time.
0 commit comments