fix: Add encryption setting to tar_and_upload_dir method (aws#3082)

Co-authored-by: Navin Soni <[email protected]>

Author: 2 people

src/sagemaker/estimator.py

@@ -16,9 +16,9 @@
 import json
 import logging
 import os
-from typing import Any, Dict
 import uuid
 from abc import ABCMeta, abstractmethod
+from typing import Any, Dict
 
 from six import string_types, with_metaclass
 from six.moves.urllib.parse import urlparse
@@ -721,6 +721,7 @@ def _stage_user_code_in_s3(self) -> str:
             dependencies=self.dependencies,
             kms_key=kms_key,
             s3_resource=self.sagemaker_session.s3_resource,
+            settings=self.sagemaker_session.settings,
         )
 
     def _prepare_rules(self):

src/sagemaker/model.py

@@ -447,6 +447,7 @@ def _upload_code(self, key_prefix: str, repack: bool = False) -> None:
                 script=self.entry_point,
                 directory=self.source_dir,
                 dependencies=self.dependencies,
+                settings=self.sagemaker_session.settings,
             )
 
         if repack and self.model_data is not None and self.entry_point is not None: