Update training_loop.py

Lightning-AI · Apr 29, 2021 · deaf510 · deaf510
1 parent 8e8b46f
commit deaf510
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py
@@ -198,7 +198,7 @@ def on_train_batch_end(self, epoch_output, batch_end_outputs, batch, batch_idx,
         # reset batch logger internals
         self.trainer.logger_connector.on_train_batch_end()
 
-    def reset_train_val_dataloaders(self, model):
+    def reset_train_val_dataloaders(self, model) -> None:
         """
         Resets train and val dataloaders if none are attached to the trainer.