From 71a4b97b2a37c4fed48ed410638a265e9a0103f4 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 26 Nov 2024 12:38:25 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../tests_fabric/strategies/test_model_parallel_integration.py | 2 +- .../tests_pytorch/strategies/test_model_parallel_integration.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/tests_fabric/strategies/test_model_parallel_integration.py b/tests/tests_fabric/strategies/test_model_parallel_integration.py index 9f7e3e67c5a52..b04a29b691529 100644 --- a/tests/tests_fabric/strategies/test_model_parallel_integration.py +++ b/tests/tests_fabric/strategies/test_model_parallel_integration.py @@ -29,7 +29,7 @@ from tests_fabric.helpers.runif import RunIf -@pytest.fixture +@pytest.fixture() def distributed(): yield if torch.distributed.is_initialized(): diff --git a/tests/tests_pytorch/strategies/test_model_parallel_integration.py b/tests/tests_pytorch/strategies/test_model_parallel_integration.py index 015b2d417a240..9dcbcc802834b 100644 --- a/tests/tests_pytorch/strategies/test_model_parallel_integration.py +++ b/tests/tests_pytorch/strategies/test_model_parallel_integration.py @@ -86,7 +86,7 @@ def fn(model, device_mesh): return fn -@pytest.fixture +@pytest.fixture() def distributed(): yield if torch.distributed.is_initialized():