diff --git a/src/rapids_pre_commit_hooks/alpha_spec.py b/src/rapids_pre_commit_hooks/alpha_spec.py index 70a97f0..07659c8 100644 --- a/src/rapids_pre_commit_hooks/alpha_spec.py +++ b/src/rapids_pre_commit_hooks/alpha_spec.py @@ -47,32 +47,7 @@ "distributed-ucxx", } -RAPIDS_CUDA_SUFFIXED_PACKAGES = { - "rmm", - "pylibcugraphops", - "pylibcugraph", - "nx-cugraph", - "dask-cudf", - "cuspatial", - "cuproj", - "cuml", - "cugraph", - "cudf", - "ptxcompiler", - "cubinlinker", - "cugraph-dgl", - "cugraph-pyg", - "cugraph-equivariant", - "raft-dask", - "pylibwholegraph", - "pylibraft", - "cuxfilter", - "cucim", - "ucx-py", - "ucxx", - "pynvjitlink", - "distributed-ucxx", -} +RAPIDS_CUDA_SUFFIXED_PACKAGES = set(RAPIDS_ALPHA_SPEC_PACKAGES) ALPHA_SPECIFIER = ">=0.0.0a0"