Why don't you try using the pathlib library?

[wakame1367]

I think we can swap out img_path.split('.')[-1].lower() for [PurePath.suffix].(https://docs.python.org/3/library/pathlib.html#pathlib.PurePath.suffix)

basic-image-eda/basic_image_eda/eda.py

Line 50 in 77cc1b3

if img_path.split('.')[-1].lower() in ['jpg', 'jpeg', 'jpe', 'png', 'bmp', 'ppm', 'pbm', 'pgm', 'sr', 'ras']:

reference_link

• Python - pathlib