refactoring

Author: Borda

pytorch_lightning/trainer/trainer.py

@@ -1023,11 +1023,10 @@ def check_testing_model_configuration(self, model: LightningModule, test_dataloa
         gave_test_loader = test_dataloader is not None
 
         if gave_test_loader and not has_test_step:
-            raise MisconfigurationException('You passed in a `test_dataloader` but did not implement '
-                                            ' `test_step()`')
+            raise MisconfigurationException('You passed in a `test_dataloader` but did not implement `test_step()`')
 
         if has_test_step and not gave_test_loader:
-            raise MisconfigurationException('You defined `test_step()` but did not implement '
+            raise MisconfigurationException('You defined `test_step()` but did not implement'
                                             ' `test_dataloader` nor passed in `.fit(test_dataloaders`.')
 
         if has_test_step and gave_test_loader and not has_test_epoch_end:

tests/base/eval_dataloaders.py tests/base/model_test_dataloaders.py

@@ -1,7 +1,7 @@
 from abc import ABC, abstractmethod
 
 
-class EvalDataloaderVariations(ABC):
+class TestDataloaderVariations(ABC):
 
     @abstractmethod
     def dataloader(self, train):

tests/base/eval_steps.py tests/base/model_test_steps.py

@@ -1,9 +1,10 @@
 from abc import ABC
 from collections import OrderedDict
+
 import torch
 
 
-class EvalStepVariations(ABC):
+class TestStepVariations(ABC):
     """
     Houses all variations of test steps
     """

tests/base/train_dataloaders.py tests/base/model_train_dataloaders.py

@@ -1,5 +1,6 @@
 from abc import ABC, abstractmethod
 
+
 class TrainDataloaderVariations(ABC):
 
     @abstractmethod

tests/base/train_steps.py tests/base/model_train_steps.py

@@ -1,5 +1,6 @@
 from abc import ABC, abstractmethod
 
+
 class ValDataloaderVariations(ABC):
 
     @abstractmethod

tests/base/valid_dataloaders.py tests/base/model_valid_dataloaders.py

@@ -3,7 +3,7 @@
 import torch
 
 
-class EvalEpochEndVariations(ABC):
+class TestEpochEndVariations(ABC):
 
     def test_epoch_end(self, outputs):
         """