def collate_fn()

in src/pixparse/task/task_cruller_eval_docvqa.py [0:0]

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

    def collate_fn(self, batch):
        question_ids = []
        image_ids = []
        images = []
        questions = []
        answers = []
        for item in batch:
            question_ids.append(item['question_id'])
            image_ids.append(item['image_id'])
            images.append(item['image'])
            questions.append(item['labels']["question"])
            answers.append(item['labels']["answers"])
   
        transform = self.image_preprocess_eval
        images = torch.stack([transform(img) for img in images])
    
        return {
            "images": images,
            "questions": questions,
            "ground_truth_answers": answers,
            "image_ids": image_ids,
            "question_ids": question_ids,
        }