removed deps

Lightning-AI · Jul 25, 2019 · 104b4dc · 104b4dc
1 parent 9b44ed1
commit 104b4dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -163,7 +163,7 @@ def test_cpu_model_with_amp():
 
     model, hparams = get_model()
 
-    with pytest.raises(MisconfigurationException):
+    with pytest.raises(ModuleNotFoundError):
         run_gpu_model_test(trainer_options, model, hparams, on_gpu=False)