dask-cuda has alpha spec but no CUDA suffix

rapidsai · May 28, 2024 · f118ae8 · f118ae8
1 parent 6841fcf
commit f118ae8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git 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
@@ -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"),