fix cudf modification

rapidsai · Sep 6, 2023 · e7da424 · e7da424
1 parent 3b7b77d
commit e7da424
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -34,6 +34,8 @@ if ! rapids-is-release-build; then
     alpha_spec=',>=0.0.0a0'
 fi
 
+sed -r -i "s/cudf==(.*)\"/cudf${PACKAGE_CUDA_SUFFIX}==\1${alpha_spec}\"/g" ${pyproject_file}
+
 if [[ $PACKAGE_CUDA_SUFFIX == "-cu12" ]]; then
     sed -i "s/cupy-cuda11x/cupy-cuda12x/g" ${pyproject_file}
 fi