def get_train_dataloader()

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

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

    def get_train_dataloader(self):
        if self.batch_sampler is None:
            return super().get_train_dataloader()
            
        return DataLoader(
            self.train_dataset,
            batch_sampler=self.batch_sampler,
            collate_fn=self.data_collator,
            num_workers=self.args.dataloader_num_workers,
            pin_memory=self.args.dataloader_pin_memory,
        )