Skip to content

Commit 96b32be

Browse files
vr140Vijay Rajaram
andauthored
[tiny] Fix training_dataloader usage to be train_dataloader instead. (#2521)
Co-authored-by: Vijay Rajaram <[email protected]>
1 parent 1098a0d commit 96b32be

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

pytorch_lightning/trainer/trainer.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1274,17 +1274,17 @@ def check_model_configuration(self, model: LightningModule):
12741274
if not self.is_overridden('training_step', model):
12751275
raise MisconfigurationException(
12761276
'No `training_step()` method defined. Lightning `Trainer` expects as minimum a'
1277-
' `training_step()`, `training_dataloader()` and `configure_optimizers()` to be defined.')
1277+
' `training_step()`, `train_dataloader()` and `configure_optimizers()` to be defined.')
12781278

12791279
if not self.is_overridden('train_dataloader', model):
12801280
raise MisconfigurationException(
12811281
'No `train_dataloader()` method defined. Lightning `Trainer` expects as minimum a'
1282-
' `training_step()`, `training_dataloader()` and `configure_optimizers()` to be defined.')
1282+
' `training_step()`, `train_dataloader()` and `configure_optimizers()` to be defined.')
12831283

12841284
if not self.is_overridden('configure_optimizers', model):
12851285
raise MisconfigurationException(
12861286
'No `configure_optimizers()` method defined. Lightning `Trainer` expects as minimum a'
1287-
' `training_step()`, `training_dataloader()` and `configure_optimizers()` to be defined.')
1287+
' `training_step()`, `train_dataloader()` and `configure_optimizers()` to be defined.')
12881288

12891289
# Check val_dataloader, validation_step and validation_epoch_end
12901290
if self.is_overridden('val_dataloader', model):

tests/trainer/test_checks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
def test_wrong_train_setting(tmpdir):
1313
"""
14-
* Test that an error is thrown when no `training_dataloader()` is defined
14+
* Test that an error is thrown when no `train_dataloader()` is defined
1515
* Test that an error is thrown when no `training_step()` is defined
1616
"""
1717
tutils.reset_seed()

0 commit comments

Comments
 (0)