update flaky syntax (#21156)

Borda · lantiga · commit 69f86083eb94 · 2025-09-05T15:14:02.000+02:00
(cherry picked from commit bb4769f)
diff --git 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
@@ -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:

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@ def spike_detection_test(fabric, global_rank_spike, spike_value, should_raise):`
`30`	`30`	`)`
`31`	`31`
`32`	`32`
`33`		`-@pytest.mark.flaky(max_runs=3)`
	`33`	`+@pytest.mark.flaky(reruns=3)`
`34`	`34`	`@pytest.mark.parametrize(`
`35`	`35`	`("global_rank_spike", "num_devices", "spike_value", "finite_only"),`
`36`	`36`	`# NOTE FOR ALL FOLLOWING TESTS:`