Skip to content

Commit 248e695

Browse files
author
Darren Stack
committed
fix: Use sagemaker session's s3 resource for downloading folders
1 parent 533f30a commit 248e695

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/sagemaker/utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -384,8 +384,7 @@ def download_folder(bucket_name, prefix, target, sagemaker_session):
384384
sagemaker_session (sagemaker.session.Session): a sagemaker session to
385385
interact with S3.
386386
"""
387-
boto_session = sagemaker_session.boto_session
388-
s3 = boto_session.resource("s3", region_name=boto_session.region_name)
387+
s3 = sagemaker_session.s3_resource
389388

390389
prefix = prefix.lstrip("/")
391390

0 commit comments

Comments
 (0)