def METHOD_NAME(
    self,
    rank: int,
    world_size: int,
) -> data_utils.DataLoader:
    sampler = DistributedSampler(
        Bert4RecTrainDataset(
            self.val,
        ),
        num_replicas=world_size,
        rank=rank,
        shuffle=False,
        drop_last=False,
    )
    dataloader = data_utils.DataLoader(
        BertEvalDataset(
            self.val,
        ),
        batch_size=self.val_batch_size,
        pin_memory=True,
        sampler=sampler,
    )
    return dataloader