def feature_extractor()

in distilvit/_datasets/coco.py [0:0]

• 11 lines of code
• 4 McCabe index (conditional complexity)

    def feature_extractor(self, *args, **kw):
        images = []
        for image_path in kw["images"]:
            try:
                images.append(Image.open(image_path).convert("RGB"))
            except Exception:
                pass

        inputs = self.base_feature_extractor(images=images, return_tensors="pt")
        for image in images:
            image.close()
        return inputs