From f118ae8dbc6de8c2e7707a0a48d1e97fcea62c11 Mon Sep 17 00:00:00 2001 From: Kyle Edwards Date: Tue, 28 May 2024 16:19:44 -0400 Subject: [PATCH] dask-cuda has alpha spec but no CUDA suffix --- src/rapids_pre_commit_hooks/alpha_spec.py | 9 ++++++++- test/rapids_pre_commit_hooks/test_alpha_spec.py | 6 ++---- 2 files changed, 10 insertions(+), 5 deletions(-) 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"),