fix

Lightning-AI · Dec 21, 2020 · 24c0cc9 · 24c0cc9
1 parent 1606878
commit 24c0cc9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/deprecated_api/test_remove_1-3.py b/tests/deprecated_api/test_remove_1-3.py
@@ -21,6 +21,7 @@
 from pytorch_lightning import LightningModule, Trainer
 from pytorch_lightning.callbacks import EarlyStopping, ModelCheckpoint
 from pytorch_lightning.profiler.profilers import PassThroughProfiler, SimpleProfiler
+from tests.deprecated_api import _soft_unimport_module
 
 
 def test_v1_3_0_deprecated_arguments(tmpdir):
@@ -137,14 +138,18 @@ def test_v1_3_0_trainer_cli_profiler(cli_args, expected_parsed_arg, expected_pro
 
 
 def test_v1_3_0_deprecated_imports():
+    _soft_unimport_module('pytorch_lightning.utilities.argparse_utils')
     with pytest.deprecated_call(match='will be removed in v1.3'):
         from pytorch_lightning.utilities.argparse_utils import from_argparse_args  # noqa: F811
 
+    _soft_unimport_module('pytorch_lightning.utilities.model_utils')
     with pytest.deprecated_call(match='will be removed in v1.3'):
         from pytorch_lightning.utilities.model_utils import is_overridden  # noqa: F811
 
+    _soft_unimport_module('pytorch_lightning.utilities.warning_utils')
     with pytest.deprecated_call(match='will be removed in v1.3'):
         from pytorch_lightning.utilities.warning_utils import WarningCache  # noqa: F811
 
+    _soft_unimport_module('pytorch_lightning.utilities.xla_device_utils')
     with pytest.deprecated_call(match='will be removed in v1.3'):
         from pytorch_lightning.utilities.xla_device_utils import XLADeviceUtils  # noqa: F811