From 2dbffab75f118d8761036d50c1d6281c69c5df3a Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 3 Aug 2021 21:29:28 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- pytorch_lightning/core/memory.py | 4 ++-- tests/deprecated_api/test_remove_1-7.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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():