fix: Use sagemaker session's s3 resource for downloading folders

Author: DarrenStack

src/sagemaker/utils.py

@@ -384,17 +384,14 @@ 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)
-
     prefix = prefix.lstrip("/")
 
     # Try to download the prefix as an object first, in case it is a file and not a 'directory'.
     # Do this first, in case the object has broader permissions than the bucket.
     if not prefix.endswith("/"):
         try:
             file_destination = os.path.join(target, os.path.basename(prefix))
-            s3.Object(bucket_name, prefix).download_file(file_destination)
+            sagemaker_session.s3_resource.Object(bucket_name, prefix).download_file(file_destination)
             return
         except botocore.exceptions.ClientError as e:
             err_info = e.response["Error"]
@@ -405,7 +402,7 @@ def download_folder(bucket_name, prefix, target, sagemaker_session):
             else:
                 raise
 
-    _download_files_under_prefix(bucket_name, prefix, target, s3)
+    _download_files_under_prefix(bucket_name, prefix, target, sagemaker_session.s3_resource)
 
 
 def _download_files_under_prefix(bucket_name, prefix, target, s3):