from pathlib import Path


def get_image_paths(image_folder: Path) -> list:
    """
    Get all images in image folder ending in .jpg or .png
    """
    suffix_set = {'.jpg', '.png'}
    return [
        image for image in image_folder.iterdir() if image.suffix in suffix_set
    ]