Skip to content

Commit

Permalink
chore(loaders): use rglob_multiple in pmdb loader
Browse files Browse the repository at this point in the history
  • Loading branch information
ndido98 committed Jan 9, 2023
1 parent 85be46f commit 72c486f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/revelio/dataset/loaders/loaders.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ def load(self, path: Path) -> list[DatasetElementDescriptor]:
y=ElementClass.BONA_FIDE,
)
)
morphed_images = sorted(path.rglob("morph*.png"))
morphed_images = rglob_multiple(path, "morph*", IMAGE_EXTENSIONS)
# Take at most n morphed images
for _, g in itertools.groupby(morphed_images, lambda x: x.stem[:-4]):
group_images = list(g)
Expand Down

0 comments on commit 72c486f

Please sign in to comment.