Update Deprecation Message to utilities/memory.py

Lightning-AI · Jul 28, 2021 · ec3240b · ec3240b
1 parent c4ace84
commit ec3240b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py
@@ -1964,7 +1964,7 @@ def model_size(self) -> float:
         """
         rank_zero_deprecation(
             'The `LightningModule.model_size` property was deprecated in v1.5 and will be removed in v1.7. Please '
-            'use the get_model_size method under utilities/model_helpers.py'
+            'use the get_model_size_mb method under utilities/memory.py'
         )
 
         tmp_name = f"{uuid.uuid4().hex}.pt"

diff --git a/tests/deprecated_api/test_remove_1-7.py b/tests/deprecated_api/test_remove_1-7.py
@@ -21,6 +21,6 @@ def test_v1_7_0_deprecated_model_size():
     model = BoringModel()
     with pytest.deprecated_call(
         match="The `LightningModule.model_size` property was deprecated in v1.5 and will be removed in v1.7. "
-        "Please use the get_model_size method under utilities/model_helpers.py"
+        "Please use the get_model_size_mb method under utilities/memory.py"
     ):
         _ = model.model_size