diff --git a/dependencies.yaml b/dependencies.yaml index 685d0a8fac..3daa400444 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -123,6 +123,7 @@ files: table: project includes: - run_raft_dask + - depends_on_ucx_run py_test_raft_dask: output: pyproject pyproject_dir: python/raft-dask @@ -132,7 +133,6 @@ files: includes: - test_python_common - depends_on_distributed_ucxx - - depends_on_ucx_run py_build_raft_ann_bench: output: pyproject pyproject_dir: python/raft-ann-bench diff --git a/python/raft-dask/pyproject.toml b/python/raft-dask/pyproject.toml index 53b1c51e83..37691cc01b 100644 --- a/python/raft-dask/pyproject.toml +++ b/python/raft-dask/pyproject.toml @@ -38,6 +38,7 @@ dependencies = [ "dask-cuda==24.6.*", "joblib>=0.11", "libraft==24.6.*", + "libucx>=1.15.0", "numba>=0.57", "numpy>=1.23,<2.0a0", "pylibraft==24.6.*", @@ -55,7 +56,6 @@ classifiers = [ [project.optional-dependencies] test = [ "distributed-ucxx==0.38.*", - "libucx>=1.15.0", "pytest-cov", "pytest==7.*", ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.