Skip to content

Commit b458d3d

Browse files
Shivakishore14laurenyu
authored andcommitted
skip object download if its a folder [Bug fix] (Local mode) (#252)
1 parent 555cfe7 commit b458d3d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/sagemaker/local/image.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -262,6 +262,9 @@ def _download_folder(self, bucket_name, prefix, target):
262262
bucket = s3.Bucket(bucket_name)
263263

264264
for obj_sum in bucket.objects.filter(Prefix=prefix):
265+
# if obj_sum is a folder object skip it.
266+
if obj_sum.key != '' and obj_sum.key[-1] == '/':
267+
continue
265268
obj = s3.Object(obj_sum.bucket_name, obj_sum.key)
266269
s3_relative_path = obj_sum.key[len(prefix):].lstrip('/')
267270
file_path = os.path.join(target, s3_relative_path)

0 commit comments

Comments
 (0)