diff --git a/ci/test_wheel_raft_dask.sh b/ci/test_wheel_raft_dask.sh index 9f8e8ce02a..676d642de9 100755 --- a/ci/test_wheel_raft_dask.sh +++ b/ci/test_wheel_raft_dask.sh @@ -12,7 +12,7 @@ RAPIDS_PY_WHEEL_NAME="pylibraft_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels python -m pip install --no-deps ./local-pylibraft-dep/pylibraft*.whl # Always install latest dask for testing -python -m pip install git+https://github.com/dask/dask.git@2023.7.1 git+https://github.com/dask/distributed.git@2023.7.1 git+https://github.com/rapidsai/dask-cuda.git@branch-23.10 +python -m pip install git+https://github.com/dask/dask.git@main git+https://github.com/dask/distributed.git@main git+https://github.com/rapidsai/dask-cuda.git@branch-23.10 # echo to expand wildcard before adding `[extra]` requires for pip python -m pip install $(echo ./dist/raft_dask*.whl)[test] diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index 223bafe70b..7e921decd5 100644 --- a/conda/environments/all_cuda-118_arch-x86_64.yaml +++ b/conda/environments/all_cuda-118_arch-x86_64.yaml @@ -19,10 +19,10 @@ dependencies: - cupy>=12.0.0 - cxx-compiler - cython>=3.0.0 -- dask-core==2023.7.1 +- dask-core>=2023.7.1 - dask-cuda==23.10.* -- dask==2023.7.1 -- distributed==2023.7.1 +- dask>=2023.7.1 +- distributed>=2023.7.1 - doxygen>=1.8.20 - gcc_linux-64=11.* - gmock>=1.13.0 diff --git a/conda/environments/all_cuda-120_arch-x86_64.yaml b/conda/environments/all_cuda-120_arch-x86_64.yaml index e68feaad82..2ea685b529 100644 --- a/conda/environments/all_cuda-120_arch-x86_64.yaml +++ b/conda/environments/all_cuda-120_arch-x86_64.yaml @@ -19,10 +19,10 @@ dependencies: - cupy>=12.0.0 - cxx-compiler - cython>=3.0.0 -- dask-core==2023.7.1 +- dask-core>=2023.7.1 - dask-cuda==23.10.* -- dask==2023.7.1 -- distributed==2023.7.1 +- dask>=2023.7.1 +- distributed>=2023.7.1 - doxygen>=1.8.20 - gcc_linux-64=11.* - gmock>=1.13.0 diff --git a/conda/recipes/raft-dask/meta.yaml b/conda/recipes/raft-dask/meta.yaml index cf1f8488bc..c9caa4dd9b 100644 --- a/conda/recipes/raft-dask/meta.yaml +++ b/conda/recipes/raft-dask/meta.yaml @@ -60,10 +60,10 @@ requirements: - cudatoolkit {% endif %} - {{ pin_compatible('cuda-version', max_pin='x', min_pin='x') }} - - dask ==2023.7.1 - - dask-core ==2023.7.1 + - dask >=2023.7.1 + - dask-core >=2023.7.1 - dask-cuda ={{ minor_version }} - - distributed ==2023.7.1 + - distributed >=2023.7.1 - joblib >=0.11 - nccl >=2.9.9 - pylibraft {{ version }} diff --git a/dependencies.yaml b/dependencies.yaml index ee04c886d7..cf8170b9a1 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -305,16 +305,16 @@ dependencies: common: - output_types: [conda, pyproject] packages: - - dask==2023.7.1 + - dask>=2023.7.1 - dask-cuda==23.10.* - - distributed==2023.7.1 + - distributed>=2023.7.1 - joblib>=0.11 - numba>=0.57 - *numpy - ucx-py==0.34.* - output_types: conda packages: - - dask-core==2023.7.1 + - dask-core>=2023.7.1 - ucx>=1.13.0 - ucx-proc=*=gpu - output_types: pyproject diff --git a/python/raft-dask/pyproject.toml b/python/raft-dask/pyproject.toml index 3c81b6f16b..bdbcf61e0f 100644 --- a/python/raft-dask/pyproject.toml +++ b/python/raft-dask/pyproject.toml @@ -35,8 +35,8 @@ license = { text = "Apache 2.0" } requires-python = ">=3.9" dependencies = [ "dask-cuda==23.10.*", - "dask==2023.7.1", - "distributed==2023.7.1", + "dask>=2023.7.1", + "distributed>=2023.7.1", "joblib>=0.11", "numba>=0.57", "numpy>=1.21",