diff --git a/ci/cpu/prebuild.sh b/ci/cpu/prebuild.sh index 746c0005a47..8a2c9d9be7c 100755 --- a/ci/cpu/prebuild.sh +++ b/ci/cpu/prebuild.sh @@ -4,12 +4,13 @@ set -e DEFAULT_CUDA_VER="11.5" +DEFAULT_PYTHON_VER="3.8" #Always upload cudf Python package export UPLOAD_CUDF=1 #Upload libcudf once per CUDA -if [[ "$PYTHON" == "3.7" ]]; then +if [[ "$PYTHON" == "${DEFAULT_PYTHON_VER}" ]]; then export UPLOAD_LIBCUDF=1 else export UPLOAD_LIBCUDF=0 @@ -23,7 +24,7 @@ else fi #We only want to upload libcudf_kafka once per python/CUDA combo -if [[ "$PYTHON" == "3.7" ]] && [[ "$CUDA" == "${DEFAULT_CUDA_VER}" ]]; then +if [[ "$PYTHON" == "${DEFAULT_PYTHON_VER}" ]] && [[ "$CUDA" == "${DEFAULT_CUDA_VER}" ]]; then export UPLOAD_LIBCUDF_KAFKA=1 else export UPLOAD_LIBCUDF_KAFKA=0