diff --git a/tools/rapids-generate-pip-constraints b/tools/rapids-generate-pip-constraints index 255fb4c..4062ec6 100755 --- a/tools/rapids-generate-pip-constraints +++ b/tools/rapids-generate-pip-constraints @@ -9,7 +9,7 @@ # Positional arguments: # # 1) file key to be passed to `rapids-dependency-file-generator --file-key` -# 2) filepath to write the constraint data to +# 2) filepath to append the constraint data to (creates it if it doesn't exist) # # [usage] # @@ -23,14 +23,8 @@ export RAPIDS_SCRIPT_NAME="rapids-generate-pip-constraints" file_key="${1}" out_file="${2}" -# 'latest' should be a no-op and not constrain dependencies at all -# (pip will prefer the latest versions of dependencies by default) -if [[ "${RAPIDS_DEPENDENCIES}" == "latest" ]]; then - echo "" > "${out_file}" -else - rapids-dependency-file-generator \ - --output requirements \ - --file-key "${file_key}" \ - --matrix "cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION};dependencies=${RAPIDS_DEPENDENCIES}" \ - | tee "${out_file}" -fi +rapids-dependency-file-generator \ + --output requirements \ + --file-key "${file_key}" \ + --matrix "cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION};dependencies=${RAPIDS_DEPENDENCIES};use_cuda_wheels=true" \ +| tee -a "${out_file}"