From f7e047e14ce1c90329b3535818a9c38ab34f3ef5 Mon Sep 17 00:00:00 2001 From: Kyle Edwards Date: Fri, 24 May 2024 11:41:05 -0400 Subject: [PATCH] Add more alpha spec packages --- src/rapids_pre_commit_hooks/alpha_spec.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/rapids_pre_commit_hooks/alpha_spec.py b/src/rapids_pre_commit_hooks/alpha_spec.py index 07659c8..298f076 100644 --- a/src/rapids_pre_commit_hooks/alpha_spec.py +++ b/src/rapids_pre_commit_hooks/alpha_spec.py @@ -45,6 +45,8 @@ "ucxx", "pynvjitlink", "distributed-ucxx", + "librmm", + "libucx", } RAPIDS_CUDA_SUFFIXED_PACKAGES = set(RAPIDS_ALPHA_SPEC_PACKAGES)