diff --git a/pytorch_lightning/testing_models/lm_test_module.py b/pytorch_lightning/testing_models/lm_test_module.py index a143e51e7e..87de2c7777 100644 --- a/pytorch_lightning/testing_models/lm_test_module.py +++ b/pytorch_lightning/testing_models/lm_test_module.py @@ -181,7 +181,9 @@ class LightningTestModel(LightningModule): """ # try no scheduler for this model (testing purposes) optimizer = optim.Adam(self.parameters(), lr=self.hparams.learning_rate) - return [optimizer], [] + + # test returning only 1 list instead of 2 + return [optimizer] def __dataloader(self, train): # init data generators