Update on_train_batch_end signature in preparation for Lightning 1.7 breaking changes

Reviewed By: wat3rBro

Differential Revision: D34813614

fbshipit-source-id: e062cf5e1cbee09bb1970a21d21bc4132aec0890

d2go/runner/lightning_task.py

@@ -425,7 +425,7 @@ class DefaultTask(pl.LightningModule):
                 device=self.cfg.MODEL_EMA.DEVICE or self.cfg.MODEL.DEVICE,
             )
 
-    def on_train_batch_end(self, outputs, batch, batch_idx, dataloader_idx) -> None:
+    def on_train_batch_end(self, *_) -> None:
         if self.ema_state:
             self.ema_state.update(self.model)