Update ci/build_wheel.sh

Co-authored-by: Vyas Ramasubramani <[email protected]>
rapidsai · Sep 8, 2023 · ce769ed · ce769ed
1 parent 8d3aa05
commit ce769ed
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -34,11 +34,9 @@ if ! rapids-is-release-build; then
     alpha_spec=',>=0.0.0a0'
 fi
 
+sed -r -i "s/rmm(.*)\"/rmm${PACKAGE_CUDA_SUFFIX}\1${alpha_spec}\"/g" ${pyproject_file}
 if [[ ${package_name} == "cuspatial" ]]; then
-    sed -r -i "s/rmm(.*)\"/rmm${PACKAGE_CUDA_SUFFIX}\1${alpha_spec}\"/g" ${pyproject_file}
     sed -r -i "s/cudf==(.*)\"/cudf${PACKAGE_CUDA_SUFFIX}==\1${alpha_spec}\"/g" ${pyproject_file}
-else
-    sed -r -i "s/rmm(.*)\"/rmm${PACKAGE_CUDA_SUFFIX}\1${alpha_spec}\"/g" ${pyproject_file}
 fi
 
 if [[ $PACKAGE_CUDA_SUFFIX == "-cu12" ]]; then