From 66c1bab1dd72029f601ed6907f9544d3bbcc9890 Mon Sep 17 00:00:00 2001 From: Jirka B Date: Wed, 3 Sep 2025 18:03:15 +0200 Subject: [PATCH] update flaky --- tests/tests_fabric/utilities/test_spike.py | 2 +- tests/tests_pytorch/callbacks/test_spike.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/tests_fabric/utilities/test_spike.py b/tests/tests_fabric/utilities/test_spike.py index e96a5f77df384..e7f0bedb8e9e9 100644 --- a/tests/tests_fabric/utilities/test_spike.py +++ b/tests/tests_fabric/utilities/test_spike.py @@ -30,7 +30,7 @@ def spike_detection_test(fabric, global_rank_spike, spike_value, should_raise): ) -@pytest.mark.flaky(max_runs=3) +@pytest.mark.flaky(reruns=3) @pytest.mark.parametrize( ("global_rank_spike", "num_devices", "spike_value", "finite_only"), # NOTE FOR ALL FOLLOWING TESTS: diff --git a/tests/tests_pytorch/callbacks/test_spike.py b/tests/tests_pytorch/callbacks/test_spike.py index f61a6c59ca9db..86e3ac88e93cf 100644 --- a/tests/tests_pytorch/callbacks/test_spike.py +++ b/tests/tests_pytorch/callbacks/test_spike.py @@ -48,7 +48,7 @@ def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx): super().on_train_batch_end(trainer, pl_module, outputs, batch, batch_idx) -@pytest.mark.flaky(max_runs=3) +@pytest.mark.flaky(reruns=3) @pytest.mark.parametrize( ("global_rank_spike", "num_devices", "spike_value", "finite_only"), # NOTE FOR ALL FOLLOWING TESTS: