diff --git a/ci/test_python.sh b/ci/test_python.sh index c42fb613..c3e7acdf 100755 --- a/ci/test_python.sh +++ b/ci/test_python.sh @@ -9,6 +9,8 @@ cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../ rapids-logger "Create test conda environment" . /opt/conda/etc/profile.d/conda.sh +UCX_PY_VERSION="$(head -1 ./VERSION)" + rapids-dependency-file-generator \ --output conda \ --file-key test_python \ @@ -50,7 +52,7 @@ PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python) rapids-mamba-retry install \ --channel "${PYTHON_CHANNEL}" \ - ucx-py + "ucx-py=${UCX_PY_VERSION}" rapids-logger "Run tests with conda package" run_tests