def METHOD_NAME(self, data):
    """ Sample data according to rank/world_size/num_workers
        Args:
            data(List): input data list
        Returns:
            List: data list after sample
    """
    data = list(range(len(data)))
    if self.partition:
        if self.shuffle:
            random.Random(self.epoch).shuffle(data)
        data = data[self.rank::self.world_size]
    data = data[self.worker_id::self.num_workers]
    return data