Skip to content

Commit ffaab34

Browse files
committed
Fix pytest-rerunfailures plugin installation and flaky marker registration
- Register 'flaky' marker in pytest.ini to eliminate warnings about unknown mark - Fix dependency group references in cuda_core/pyproject.toml to use {include-group = "test"} instead of "cuda-core[test]" extra - This ensures pytest-rerunfailures is properly installed so @pytest.mark.flaky() decorators work correctly Example failure fixed by this commit: https://github.com/NVIDIA/cuda-python/actions/runs/22501839950/job/65191834971?pr=1702 Made-with: Cursor
1 parent 3ed5f0f commit ffaab34

2 files changed

Lines changed: 5 additions & 4 deletions

File tree

cuda_core/pyproject.toml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -57,12 +57,12 @@ cu13 = ["cuda-bindings[all]==13.*"]
5757
[dependency-groups]
5858
test = ["cython>=3.2,<3.3", "setuptools", "pytest>=6.2.4", "pytest-randomly", "pytest-repeat", "pytest-rerunfailures"]
5959
ml-dtypes = ["ml-dtypes>=0.5.4,<0.6.0"]
60-
test-cu12 = [ {include-group = "ml-dtypes" }, "cuda-core[test]", "cupy-cuda12x; python_version < '3.14'", "cuda-toolkit[cudart]==12.*"] # runtime headers needed by CuPy
61-
test-cu13 = [ {include-group = "ml-dtypes" }, "cuda-core[test]", "cupy-cuda13x; python_version < '3.14'", "cuda-toolkit[cudart]==13.*"] # runtime headers needed by CuPy
60+
test-cu12 = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cupy-cuda12x; python_version < '3.14'", "cuda-toolkit[cudart]==12.*"] # runtime headers needed by CuPy
61+
test-cu13 = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cupy-cuda13x; python_version < '3.14'", "cuda-toolkit[cudart]==13.*"] # runtime headers needed by CuPy
6262
# free threaded build, cupy doesn't support free-threaded builds yet, so avoid installing it for now
6363
# TODO: cupy should support free threaded builds
64-
test-cu12-ft = [ {include-group = "ml-dtypes" }, "cuda-core[test]", "cuda-toolkit[cudart]==12.*"]
65-
test-cu13-ft = [ {include-group = "ml-dtypes" }, "cuda-core[test]", "cuda-toolkit[cudart]==13.*"]
64+
test-cu12-ft = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cuda-toolkit[cudart]==12.*"]
65+
test-cu13-ft = [ {include-group = "ml-dtypes" }, {include-group = "test" }, "cuda-toolkit[cudart]==13.*"]
6666

6767
[project.urls]
6868
homepage = "https://nvidia.github.io/cuda-python/"

pytest.ini

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,3 +19,4 @@ markers =
1919
core: tests for cuda_core
2020
cython: cython tests
2121
smoke: meta-level smoke tests
22+
flaky: mark test as flaky (provided by pytest-rerunfailures)

0 commit comments

Comments
 (0)