diff --git a/ci/build_python_pypi.sh b/ci/build_python_pypi.sh index b13783d16..66bc7cf2e 100755 --- a/ci/build_python_pypi.sh +++ b/ci/build_python_pypi.sh @@ -17,6 +17,16 @@ if ! rapids-is-release-build; then export PACKAGE_VERSION_NUMBER="${version}" fi +# For nightlies we want to ensure that we're pulling in alphas as well. The +# easiest way to do so is to augment the spec with a constraint containing a +# min alpha version that doesn't affect the version bounds but does allow usage +# of alpha versions for that dependency without --pre +alpha_spec='' +if ! rapids-is-release-build; then + alpha_spec=',>=0.0.0a0' +fi + +sed -r -i "s/rapids-dask-dependency==(.*)\"/rapids-dask-dependency==\1${alpha_spec}\"/g" pyproject.toml echo "${version}" | tr -d '"' > VERSION sed -i "/^__git_commit__/ s/= .*/= \"${commit}\"/g" "${package_name}/_version.py"