more fixes to CI scripts

rapidsai · Dec 17, 2024 · f492d59 · f492d59
1 parent 97f5d31
commit f492d59
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -40,6 +40,12 @@ case "${RAPIDS_CUDA_VERSION}" in
   ;;
 esac
 
+if [[ ${package_name} != "libraft" ]]; then
+    EXCLUDE_ARGS+=(
+      --exclude "libraft.so"
+    )
+fi
+
 sccache --zero-stats
 
 rapids-logger "Building '${package_name}' wheel"

diff --git a/ci/validate_wheel.sh b/ci/validate_wheel.sh
@@ -21,7 +21,7 @@ if [[ "${package_name}" == "libraft" ]]; then
             --max-allowed-size-compressed '5.0G'
         )
     fi
-elif [[ "${package_name}" == "pylibraft" ]]; then
+elif [[ "${package_name}" == "pylibraft" ]] || [[ "${package_name}" == "raft-dask" ]]; then
     if [[ "${RAPIDS_CUDA_MAJOR}" == "11" ]]; then
         PYDISTCHECK_ARGS=(
             --max-allowed-size-compressed '600M'