def __iter__()

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

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

    def __iter__(self):
        indexes = [
            idx
            for idx in torch.randperm(self.num_samples).tolist()
            for _ in range(self.repeat_count)
        ]
        return iter(indexes)