From 104b4dc1ff1ce7eb016a5d268bf34aa7a5eac800 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 21:28:34 -0400
Subject: [PATCH] removed deps

---
 tests/test_models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_models.py b/tests/test_models.py
index 552739eb52..fcbcbce635 100644
--- 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)