def has_valid_image_size()

in src/open-r1-multimodal/local_scripts/prepare_hf_data.py [0:0]

• 8 lines of code
• 10 McCabe index (conditional complexity)

def has_valid_image_size(example): # for Qwen2-VL-2B's processor requirement
    # Assuming the image is in a format that can be checked for dimensions
    # You might need to adjust this depending on how the image is stored in your dataset
    try:
        image = example["image"]  # or however your image is accessed
        if isinstance(image, dict) and "height" in image and "width" in image:
            return image["height"] >= 28 and image["width"] >= 28
        # If image is a PIL Image or similar
        return image.height >= 28 and image.width >= 28
    except:
        return False