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

aws · May 9, 2024 · 248e695 · 248e695
1 parent 533f30a
commit 248e695
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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("/")