fix test

Lightning-AI · Jun 28, 2020 · c629210 · c629210
1 parent 6d0248b
commit c629210
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/models/test_hparams.py b/tests/models/test_hparams.py
@@ -459,10 +459,11 @@ def configure_optimizers(self):
 
 
 @pytest.mark.parametrize("cls", [
-    NoArgsSubClassEvalModel,
+    SimpleNoArgsModel,
     NoArgsSubClassEvalModel,
 ])
-def test_nohparams_train_test(tmpdir, cls):
+def test_model_nohparams_train_test(tmpdir, cls):
+    """Test models that do not tae any argument in init."""
 
     model = cls()
     trainer = Trainer(
@@ -478,7 +479,7 @@ def test_nohparams_train_test(tmpdir, cls):
 
 
 def test_model_ignores_non_exist_kwargument(tmpdir):
-    """ Test that the model takes nly valid class arguments """
+    """Test that the model takes only valid class arguments."""
 
     class LocalModel(EvalModelTemplate):
         def __init__(self, batch_size=15):