File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 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
2525rank_zero_deprecation (
2626 "`pytorch_lightning.core.memory.LayerSummary` and"
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
Original file line number Diff line number Diff line change @@ -31,13 +31,13 @@ def test_v1_7_0_deprecated_lightning_module_summarize(tmpdir):
3131def 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
3737def 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
4343def test_v1_7_0_deprecated_model_size ():
You can’t perform that action at this time.
0 commit comments