From 0950b07a328809335470812959341c585f1a9e2a Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 10 Apr 2025 09:32:53 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- tests/test_booster/test_mixed_precision/test_fp16_torch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_booster/test_mixed_precision/test_fp16_torch.py b/tests/test_booster/test_mixed_precision/test_fp16_torch.py index 3fd6b7df1..f6d6e8303 100644 --- a/tests/test_booster/test_mixed_precision/test_fp16_torch.py +++ b/tests/test_booster/test_mixed_precision/test_fp16_torch.py @@ -34,6 +34,7 @@ def run_torch_amp(rank, world_size, port): optimizer.step() del model, optimizer, criterion, data, output, mixed_precision + @rerun_if_address_is_in_use() def test_torch_ddp_plugin(): spawn(run_torch_amp, 1)