Fix test

Lightning-AI · Mar 29, 2021 · b15434d · b15434d
1 parent 137dd04
commit b15434d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
@@ -1782,7 +1782,7 @@ def on_fit_start(self, trainer, pl_module: LightningModule) -> None:
 def test_exception_when_testing_or_validating_with_fast_dev_run(tmpdir):
     trainer = Trainer(default_root_dir=tmpdir, fast_dev_run=True)
 
-    with pytest.raises(MisconfigurationException, match=r"trainer.validate\(\)` with `fast_dev_run=True"):
+    with pytest.raises(MisconfigurationException, match=r"\.validate\(\)` with `fast_dev_run=True"):
         trainer.validate()
-    with pytest.raises(MisconfigurationException, match=r"trainer.test\(\)` with `fast_dev_run=True"):
+    with pytest.raises(MisconfigurationException, match=r"\.test\(\)` with `fast_dev_run=True"):
         trainer.test()