def process_batch()

in distilvit/curate_api.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

    def process_batch(self, batch):
        batch["original_caption"] = list(batch["caption"])
        new_captions = []

        for caption in batch["caption"]:
            new_captions.append([self.process_caption(caption)])

        batch["caption"] = new_captions
        return batch