diff --git a/src/sagemaker/utils.py b/src/sagemaker/utils.py index 0436c0afea..b962fae15e 100644 --- a/src/sagemaker/utils.py +++ b/src/sagemaker/utils.py @@ -384,8 +384,7 @@ def download_folder(bucket_name, prefix, target, sagemaker_session): sagemaker_session (sagemaker.session.Session): a sagemaker session to interact with S3. """ - boto_session = sagemaker_session.boto_session - s3 = boto_session.resource("s3", region_name=boto_session.region_name) + s3 = sagemaker_session.s3_resource prefix = prefix.lstrip("/") @@ -657,9 +656,8 @@ def download_file(bucket_name, path, target, sagemaker_session): interact with S3. """ path = path.lstrip("/") - boto_session = sagemaker_session.boto_session - s3 = boto_session.resource("s3", region_name=sagemaker_session.boto_region_name) + s3 = sagemaker_session.s3_resource bucket = s3.Bucket(bucket_name) bucket.download_file(path, target) diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py index 81d8279e6d..76f9dce5d0 100644 --- a/tests/unit/test_utils.py +++ b/tests/unit/test_utils.py @@ -1032,10 +1032,10 @@ def __init__(self, tmp): self.current_bucket = None self.object_mock = MagicMock() - self.sagemaker_session.boto_session.resource().Bucket().download_file.side_effect = ( + self.sagemaker_session.s3_resource.Bucket().download_file.side_effect = ( self.download_file ) - self.sagemaker_session.boto_session.resource().Bucket.side_effect = self.bucket + self.sagemaker_session.s3_resource.Bucket.side_effect = self.bucket self.fake_upload_path = self.mock_s3_upload() def bucket(self, name):