Skip to content

Commit 50e4f8e

Browse files
committed
Bug fixed.
1 parent c33d2dc commit 50e4f8e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cached_augment.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,11 +11,11 @@
1111

1212
def augment_item(src: str, transform: TransformBase, branch: str = "") -> Item | list[Item]:
1313
if not isdir(target := f"{src}/{branch}"):
14+
if not target.endswith(".png"):
15+
return []
1416
return f"{branch}", transform(imread(src))
1517
r = []
1618
for f in listdir(target):
17-
if not f.endswith(".png"):
18-
continue
1919
if isinstance(item := augment_item(src, transform, f"{branch}/{f}"), list):
2020
r += item
2121
else:

0 commit comments

Comments
 (0)