pytorch-lightning

9c4a3fd6 - Always pass the correct batch index to the automatic optimization loop (#18619)