diff --git a/src/rapids_pre_commit_hooks/alpha_spec.py b/src/rapids_pre_commit_hooks/alpha_spec.py index 95515d2..aa74591 100644 --- a/src/rapids_pre_commit_hooks/alpha_spec.py +++ b/src/rapids_pre_commit_hooks/alpha_spec.py @@ -32,6 +32,7 @@ "cuproj", "cuspatial", "cuxfilter", + "dask-cuda", "dask-cudf", "distributed-ucxx", "librmm", @@ -49,7 +50,13 @@ "ucxx", } -RAPIDS_CUDA_SUFFIXED_PACKAGES = set(RAPIDS_ALPHA_SPEC_PACKAGES) +RAPIDS_NON_CUDA_SUFFIXED_PACKAGES = { + "dask-cuda", +} + +RAPIDS_CUDA_SUFFIXED_PACKAGES = ( + RAPIDS_ALPHA_SPEC_PACKAGES - RAPIDS_NON_CUDA_SUFFIXED_PACKAGES +) ALPHA_SPECIFIER = ">=0.0.0a0" diff --git a/test/rapids_pre_commit_hooks/test_alpha_spec.py b/test/rapids_pre_commit_hooks/test_alpha_spec.py index 9661cf5..425c2b4 100644 --- a/test/rapids_pre_commit_hooks/test_alpha_spec.py +++ b/test/rapids_pre_commit_hooks/test_alpha_spec.py @@ -42,8 +42,7 @@ (f"{p}-cu12", False), (f"{p}-cuda", False), ] - for p in alpha_spec.RAPIDS_ALPHA_SPEC_PACKAGES - - alpha_spec.RAPIDS_CUDA_SUFFIXED_PACKAGES + for p in alpha_spec.RAPIDS_NON_CUDA_SUFFIXED_PACKAGES ) ), ], @@ -83,8 +82,7 @@ def test_is_rapids_cuda_suffixed_package(name, is_suffixed): (f"{p}-cu12", f"{p}-cu12", "development", None), (f"{p}-cu12", f"{p}-cu12>=0.0.0a0", "release", None), ] - for p in alpha_spec.RAPIDS_ALPHA_SPEC_PACKAGES - - alpha_spec.RAPIDS_CUDA_SUFFIXED_PACKAGES + for p in alpha_spec.RAPIDS_NON_CUDA_SUFFIXED_PACKAGES ) ), ("cuml", "cuml>=24.04,<24.06", "development", "cuml>=24.04,<24.06,>=0.0.0a0"),