Also use the latest raft in the Python build

rapidsai · Dec 17, 2024 · 38b7908 · 38b7908
1 parent 1ebf9be
commit 38b7908
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ci/build_python.sh b/ci/build_python.sh
@@ -19,6 +19,7 @@ package_name="cuvs"
 package_dir="python"
 
 CPP_CHANNEL=$(rapids-download-conda-from-s3 cpp)
+LIBRAFT_CHANNEL=$(rapids-get-pr-conda-artifact raft 2530 cpp)
 
 version=$(rapids-generate-version)
 export RAPIDS_PACKAGE_VERSION=${version}
@@ -31,6 +32,7 @@ sccache --zero-stats
 rapids-conda-retry mambabuild \
   --no-test \
   --channel "${CPP_CHANNEL}" \
+    --channel "${LIBRAFT_CHANNEL}" \
   conda/recipes/cuvs
 
 sccache --show-adv-stats
@@ -40,6 +42,7 @@ sccache --zero-stats
 rapids-conda-retry mambabuild \
   --no-test \
   --channel "${CPP_CHANNEL}" \
+    --channel "${LIBRAFT_CHANNEL}" \
   --channel "${RAPIDS_CONDA_BLD_OUTPUT_DIR}" \
   conda/recipes/cuvs-bench
 
@@ -53,6 +56,7 @@ if [[ ${RAPIDS_CUDA_MAJOR} == "12" ]]; then
   rapids-conda-retry mambabuild \
   --no-test \
   --channel "${CPP_CHANNEL}" \
+    --channel "${LIBRAFT_CHANNEL}" \
   --channel "${RAPIDS_CONDA_BLD_OUTPUT_DIR}" \
   conda/recipes/cuvs-bench-cpu