def __call__()

in distilvit/utils.py [0:0]

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

    def __call__(self, examples):
        if isinstance(examples[self.caption_column], list):
            captions = [cap[0] for cap in examples[self.caption_column]]
        else:
            captions = examples[self.caption_column]

        return {
            "labels": self.tokenize(captions),
            "pixel_values": self.feature_extractor(
                images=examples[self.image_column], return_tensors="pt"
            )["pixel_values"],
        }