Add missing stage

Lightning-AI · Mar 16, 2021 · 4fc6e51 · 4fc6e51
1 parent 9395139
commit 4fc6e51
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/connectors/data_connector.py b/pytorch_lightning/trainer/connectors/data_connector.py
@@ -81,7 +81,7 @@ def attach_data(self, model, train_dataloader, val_dataloaders, datamodule):
 
         # set up the passed in dataloaders (if needed)
         self.attach_dataloaders(model, train_dataloader, val_dataloaders)
-        self.attach_datamodule(model, datamodule)
+        self.attach_datamodule(model, datamodule, 'fit')
         self._validate_data_hooks(model)
 
     def __enforce_datamodule_dataloader_override(self, train_dataloader, val_dataloaders, datamodule):