Source code for ml3d.torch.dataloaders.torch_sampler

import torch
from torch.utils.data import Sampler


class TorchSamplerWrapper(Sampler):

    def __init__(self, sampler):
        self.sampler = sampler

    def __iter__(self):
        return self.sampler.get_cloud_sampler()

    def __len__(self):
        return len(self.sampler)


def get_sampler(sampler):
    return TorchSamplerWrapper(sampler)