uniform_batch_size or the temp copy of the attributes of cls on ```partial```?