batched predict also for `PredictionSampler`

### Description of feature

Let's have the `PredictionSampler` such that we can use the batched version. Therefore, we check whether the number of control cells is constant. 