Use pytest-rerunfailures for flaky tests (#2279)

mathbunnyru · web-flow · commit 0ff54bbcb365 · 2025-04-04T02:31:14.000+01:00
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -2,7 +2,7 @@ docker
 plumbum
 pre-commit
 pytest
-pytest-retry
+pytest-rerunfailures
 # `pytest-xdist` is a plugin that provides the `--numprocesses` flag,
 # allowing us to run `pytest` tests in parallel
 pytest-xdist
diff --git a/tests/by_image/all-spark-notebook/test_spark_r_nbconvert.py b/tests/by_image/all-spark-notebook/test_spark_r_nbconvert.py
@@ -12,7 +12,7 @@
 THIS_DIR = Path(__file__).parent.resolve()
 
 
-@pytest.mark.flaky(retries=3, delay=1)
+@pytest.mark.flaky(reruns=3, reruns_delay=1)
 @pytest.mark.parametrize(
     "test_file,expected_warnings",
     [

Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@`
`12`	`12`	`THIS_DIR = Path(__file__).parent.resolve()`
`13`	`13`
`14`	`14`
`15`		`-@pytest.mark.flaky(retries=3, delay=1)`
	`15`	`+@pytest.mark.flaky(reruns=3, reruns_delay=1)`
`16`	`16`	`@pytest.mark.parametrize(`
`17`	`17`	`"test_file,expected_warnings",`
`18`	`18`	`[`