fixes #909

Lightning-AI · Feb 25, 2020 · 9504461 · 9504461
1 parent 05ad57d
commit 9504461
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_trainer.py b/tests/test_trainer.py
@@ -645,9 +645,9 @@ class CurrentTestModel(
     trainer = Trainer(**trainer_options)
     fit_options = dict(train_dataloader=model._dataloader(train=True),
                        val_dataloaders=[model._dataloader(train=False),
-                                       model._dataloader(train=False)],
+                                        model._dataloader(train=False)],
                        test_dataloaders=[model._dataloader(train=False),
-                                        model._dataloader(train=False)])
+                                         model._dataloader(train=False)])
     results = trainer.fit(model, **fit_options)
     trainer.test()
 
@@ -662,9 +662,9 @@ def test_mixing_of_dataloader_options(tmpdir):
     tutils.reset_seed()
 
     class CurrentTestModel(
-        LightningTestModelBase,
         LightningValStepFitSingleDataloaderMixin,
-        LightningTestFitSingleTestDataloadersMixin
+        LightningTestFitSingleTestDataloadersMixin,
+        LightningTestModelBase,
     ):
         pass