diff --git a/pytorch_lightning/core/memory.py b/pytorch_lightning/core/memory.py index 7957fdecf8432..d61164e6d88cb 100644 --- a/pytorch_lightning/core/memory.py +++ b/pytorch_lightning/core/memory.py @@ -20,7 +20,7 @@ ) # To support backward compatibility as get_memory_profile and get_gpu_memory_map have been moved -from pytorch_lightning.utilities.memory import get_gpu_memory_map, get_memory_profile # noqa: E402 F401 # isort: skip +from pytorch_lightning.utilities.memory import get_gpu_memory_map, get_memory_profile # noqa: E402, F401 # isort: skip rank_zero_deprecation( "`pytorch_lightning.core.memory.LayerSummary` and" @@ -29,4 +29,4 @@ ) # To support backward compatibility as LayerSummary and ModelSummary have been moved -from pytorch_lightning.utilities.model_summary import LayerSummary, ModelSummary # noqa: E402 F401 # isort: skip +from pytorch_lightning.utilities.model_summary import LayerSummary, ModelSummary # noqa: E402, F401 # isort: skip diff --git a/tests/deprecated_api/test_remove_1-7.py b/tests/deprecated_api/test_remove_1-7.py index 3be195e14bcf9..38b1e50150063 100644 --- a/tests/deprecated_api/test_remove_1-7.py +++ b/tests/deprecated_api/test_remove_1-7.py @@ -31,13 +31,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"): - from pytorch_lightning.core.memory import LayerSummary, ModelSummary # noqa: F811 F401 + from pytorch_lightning.core.memory import LayerSummary, ModelSummary # noqa: 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"): - from pytorch_lightning.core.memory import get_gpu_memory_map, get_memory_profile # noqa: F811 F401 + from pytorch_lightning.core.memory import get_gpu_memory_map, get_memory_profile # noqa: F401 def test_v1_7_0_deprecated_model_size():