From 248e695e6cb93ba27e4f7926ed02671da8f0d8f1 Mon Sep 17 00:00:00 2001
From: Darren Stack <darren@cumulusneuro.com>
Date: Thu, 9 May 2024 15:53:52 +0100
Subject: [PATCH] fix: Use sagemaker session's s3 resource for downloading
 folders

---
 src/sagemaker/utils.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/sagemaker/utils.py b/src/sagemaker/utils.py
index 0436c0afea..cab9647399 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("/")