### 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.
Description of feature
Let's have the
PredictionSamplersuch that we can use the batched version. Therefore, we check whether the number of control cells is constant.