From 64b1825a338b5124aa1939e16e245e6e442df615 Mon Sep 17 00:00:00 2001
From: Navin Soni <navinns@amazon.com>
Date: Wed, 27 Apr 2022 00:25:32 +0000
Subject: [PATCH] fix: Add encryption setting to tar_and_upload_dir method

Add encryption setting to tar_and_upload_dir method in following files
- sagemaker/estimator.py
-  sagemaker/model.py
---
 src/sagemaker/estimator.py | 3 ++-
 src/sagemaker/model.py     | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py
index 2aa8d83225..5fe1eb91ec 100644
--- a/src/sagemaker/estimator.py
+++ b/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):
diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py
index 56a00478d3..4f4ae6a3dd 100644
--- a/src/sagemaker/model.py
+++ b/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: