pytorch-lightning

438f29f0 - Relax restrictions on wrapping a custom batch sampler in predict (#19678)