def METHOD_NAME(
    self,
    dataset,
    batch_size,
    last_batch,
    collate_fn,
    sampler,
    batch_sampler,
    num_workers,
    pin_memory,
    shuffle,
    distributed,
):
    raise NotImplementedError