From 36e13e5db3ab335d32f74631aba0b29c627f936b Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 12 Jul 2024 23:35:38 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- tests/tests_pytorch/profilers/test_profiler.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/tests/tests_pytorch/profilers/test_profiler.py b/tests/tests_pytorch/profilers/test_profiler.py index ec609f3a256ee..6f92788e270d6 100644 --- a/tests/tests_pytorch/profilers/test_profiler.py +++ b/tests/tests_pytorch/profilers/test_profiler.py @@ -519,11 +519,14 @@ def __init__(self): assert "[pl][module]torch.nn.modules.linear.Linear: layer.2" in event_names -@pytest.mark.parametrize("cls", [ - SimpleProfiler, - PyTorchProfiler, - pytest.param(AdvancedProfiler, marks=skip_advanced_profiler_py312), -]) +@pytest.mark.parametrize( + "cls", + [ + SimpleProfiler, + PyTorchProfiler, + pytest.param(AdvancedProfiler, marks=skip_advanced_profiler_py312), + ], +) def test_profiler_teardown(tmp_path, cls): """This test checks if profiler teardown method is called when trainer is exiting."""