fix ticks in warning message test

Lightning-AI · Aug 3, 2021 · 9dc913a · 9dc913a
1 parent b7f47b8
commit 9dc913a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/deprecated_api/test_remove_1-7.py b/tests/deprecated_api/test_remove_1-7.py
@@ -30,13 +30,13 @@ def test_v1_7_0_deprecated_lightning_module_summarize(tmpdir):
 
 def test_v1_7_0_moved_model_summary_and_layer_summary(tmpdir):
     _soft_unimport_module("pytorch_lightning.core.memory")
-    with pytest.deprecated_call(match="to pytorch_lightning.utilities.model_summary since v1.5"):
+    with pytest.deprecated_call(match="to `pytorch_lightning.utilities.model_summary` since v1.5"):
         from pytorch_lightning.core.memory import LayerSummary, ModelSummary  # noqa: F811 F401
 
 
 def test_v1_7_0_moved_get_memory_profile_and_get_gpu_memory_map(tmpdir):
     _soft_unimport_module("pytorch_lightning.core.memory")
-    with pytest.deprecated_call(match="to pytorch_lightning.utilities.memory since v1.5"):
+    with pytest.deprecated_call(match="to `pytorch_lightning.utilities.memory` since v1.5"):
         from pytorch_lightning.core.memory import get_gpu_memory_map, get_memory_profile  # noqa: F811 F401