def METHOD_NAME(self, *args, **kwargs):
    indices = self.dataset.METHOD_NAME(*args, **kwargs)
    # Hacky solution for sampling
    size = int(self.sample * indices.shape[0])
    return indices.take(np.sort(np.random.permutation(indices.shape[0])[:size]))