From b17a3dc17159226f4430d99d18dd1fda974cd8a4 Mon Sep 17 00:00:00 2001 From: Mieszko Dziadowiec Date: Fri, 10 Jan 2025 15:34:49 +0100 Subject: [PATCH] Remove not used tmp_path argument --- tests/tests_pytorch/strategies/test_ddp_integration.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/tests_pytorch/strategies/test_ddp_integration.py b/tests/tests_pytorch/strategies/test_ddp_integration.py index 23e83e95b1d7b..bd57674782674 100644 --- a/tests/tests_pytorch/strategies/test_ddp_integration.py +++ b/tests/tests_pytorch/strategies/test_ddp_integration.py @@ -454,7 +454,7 @@ def creates_processes_externally(self): @RunIf(min_cuda_gpus=2, standalone=True, dynamo=True) @mock.patch("lightning.fabric.wrappers.torch.compile", Mock(wraps=torch.compile)) @mock.patch.dict(os.environ, {}) -def test_reapply_compile(tmp_path): +def test_reapply_compile(): """Test that Trainer can rewrap a compiled module such that compilation happens over the DDP-wrapper.""" trainer = Trainer(accelerator="gpu", devices=2, strategy="ddp", max_steps=2, logger=False)