Skip to content

Commit 2dbffab

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 9dc913a commit 2dbffab

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

pytorch_lightning/core/memory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
)
2121

2222
# To support backward compatibility as get_memory_profile and get_gpu_memory_map have been moved
23-
from pytorch_lightning.utilities.memory import get_gpu_memory_map, get_memory_profile # noqa: E402 F401 # isort: skip
23+
from pytorch_lightning.utilities.memory import get_gpu_memory_map, get_memory_profile # noqa: E402, F401 # isort: skip
2424

2525
rank_zero_deprecation(
2626
"`pytorch_lightning.core.memory.LayerSummary` and"
@@ -29,4 +29,4 @@
2929
)
3030

3131
# To support backward compatibility as LayerSummary and ModelSummary have been moved
32-
from pytorch_lightning.utilities.model_summary import LayerSummary, ModelSummary # noqa: E402 F401 # isort: skip
32+
from pytorch_lightning.utilities.model_summary import LayerSummary, ModelSummary # noqa: E402, F401 # isort: skip

tests/deprecated_api/test_remove_1-7.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,13 +31,13 @@ def test_v1_7_0_deprecated_lightning_module_summarize(tmpdir):
3131
def test_v1_7_0_moved_model_summary_and_layer_summary(tmpdir):
3232
_soft_unimport_module("pytorch_lightning.core.memory")
3333
with pytest.deprecated_call(match="to `pytorch_lightning.utilities.model_summary` since v1.5"):
34-
from pytorch_lightning.core.memory import LayerSummary, ModelSummary # noqa: F811 F401
34+
from pytorch_lightning.core.memory import LayerSummary, ModelSummary # noqa: F401
3535

3636

3737
def test_v1_7_0_moved_get_memory_profile_and_get_gpu_memory_map(tmpdir):
3838
_soft_unimport_module("pytorch_lightning.core.memory")
3939
with pytest.deprecated_call(match="to `pytorch_lightning.utilities.memory` since v1.5"):
40-
from pytorch_lightning.core.memory import get_gpu_memory_map, get_memory_profile # noqa: F811 F401
40+
from pytorch_lightning.core.memory import get_gpu_memory_map, get_memory_profile # noqa: F401
4141

4242

4343
def test_v1_7_0_deprecated_model_size():

0 commit comments

Comments
 (0)