def __iter__()

in src/open-r1-multimodal/src/open_r1/grpo_rec.py [0:0]

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

    def __iter__(self):
        # 合并所有可能的batch
        all_batches = []
        # 图像batch
        random.shuffle(self.image_indices)
        for i in range(0, len(self.image_indices), self.batch_size):
            all_batches.append(self.image_indices[i:i+self.batch_size])
        # 文本batch
        random.shuffle(self.text_indices)
        for i in range(0, len(self.text_indices), self.batch_size):
            all_batches.append(self.text_indices[i:i+self.batch_size])
        # 随机打乱batch顺序
        random.shuffle(all_batches)
        return iter(all_batches)