if (self.sampler is not None):
            task_size = len(self.sampler)
        else:
            task_size = len(self.task)