def METHOD_NAME(
    self,
    dataset: InputBase,
    batch_size: int,
    num_workers: int = 0,
    pin_memory: bool = False,
    shuffle: bool = False,
    drop_last: bool = False,
    sampler: Optional[Sampler] = None,
    persistent_workers: bool = False,
    input_transform: Optional[InputTransform] = None,
    trainer: Optional["flash.Trainer"] = None,
) -> DataLoader:
    return self.adapter.METHOD_NAME(
        dataset,
        batch_size,
        num_workers=num_workers,
        pin_memory=pin_memory,
        shuffle=shuffle,
        drop_last=drop_last,
        sampler=sampler,
        persistent_workers=persistent_workers,
        input_transform=input_transform,
        trainer=trainer,
    )