added wheel build location

Author: VenkateshJaya

.github/workflows/build.yaml

@@ -52,6 +52,8 @@ jobs:
       sha: ${{ inputs.sha }}
       date: ${{ inputs.date }}
       script: ci/build_wheel.sh
+      package-type: python
+      wheel-name: ucx_py
   wheel-publish:
     needs: wheel-build
     secrets: inherit

.github/workflows/pr.yaml

@@ -59,6 +59,8 @@ jobs:
     with:
       build_type: pull-request
       script: ci/build_wheel.sh
+      package-type: python
+      wheel-name: ucx_py
   wheel-tests:
     needs: wheel-build
     secrets: inherit

ci/build_wheel.sh

@@ -5,6 +5,8 @@ set -euo pipefail
 
 source rapids-date-string
 
+wheel_dir=${RAPIDS_WHEEL_BLD_OUTPUT_DIR}
+
 rapids-generate-version > ./VERSION
 
 RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
@@ -17,16 +19,15 @@ rapids-pip-retry wheel \
     --config-settings rapidsai.disable-cuda=false \
     .
 
-mkdir -p final_dist
 python -m auditwheel repair \
-    -w final_dist \
+    -w "${wheel_dir}" \
     --exclude "libucm.so.0" \
     --exclude "libucp.so.0" \
     --exclude "libucs.so.0" \
     --exclude "libucs_signal.so.0" \
     --exclude "libuct.so.0" \
     dist/*
 
-./ci/validate_wheel.sh final_dist
+./ci/validate_wheel.sh "${wheel_dir}"
 
-RAPIDS_PY_WHEEL_NAME="ucx_py_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 python final_dist
+RAPIDS_PY_WHEEL_NAME="ucx_py_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 python "${wheel_dir}"