Skip to content

Commit 53c4c5b

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 Made-with: Cursor
1 parent 3ed5f0f commit 53c4c5b

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)