From 4bb00ef997d81cf51ea66a9e9356e08d63590c0e Mon Sep 17 00:00:00 2001
From: Vyas Ramasubramani <vyasr@nvidia.com>
Date: Mon, 3 Feb 2025 18:28:21 +0000
Subject: [PATCH] Fix CI test download

---
 ci/build_cpp.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ci/build_cpp.sh b/ci/build_cpp.sh
index 5f4864ea..391e2dcf 100755
--- a/ci/build_cpp.sh
+++ b/ci/build_cpp.sh
@@ -20,9 +20,11 @@ rapids-logger "Begin C++ and Python builds"
 sccache --zero-stats
 
 LIBRMM_CHANNEL=$(_rapids-get-pr-artifact rmm 1808 cpp conda)
+PYLIBRMM_CHANNEL=$(_rapids-get-pr-artifact rmm 1808 python conda)
 
 rapids-conda-retry mambabuild \
   --channel "${LIBRMM_CHANNEL}" \
+  --channel "${PYLIBRMM_CHANNEL}" \
   conda/recipes/ucxx
 
 sccache --show-adv-stats