Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jul 12, 2024
1 parent f19baea commit 36e13e5
Showing 1 changed file with 8 additions and 5 deletions.
13 changes: 8 additions & 5 deletions tests/tests_pytorch/profilers/test_profiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -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."""

Expand Down

0 comments on commit 36e13e5

Please sign in to comment.