def METHOD_NAME(self):
    return self.plugin.prepare_dataloader(self.dataset["train"],
                                          batch_size=self.train_batch_size,
                                          shuffle=True,
                                          drop_last=True)