From 6b731b6df0218f75fefcf288880f39cbd934d58d Mon Sep 17 00:00:00 2001
From: Divye Gala <divyegala@gmail.com>
Date: Wed, 8 May 2024 06:51:29 -0400
Subject: [PATCH] `libucx` should be run dependency of `raft-dask` (#2296)

As figured out in https://github.com/rapidsai/cuml/pull/5697, `libucx` should not be just a testing dependency of `raft-dask`, it should be a run dependency.

cc @vyasr

Authors:
  - Divye Gala (https://github.com/divyegala)
  - Vyas Ramasubramani (https://github.com/vyasr)

Approvers:
  - Vyas Ramasubramani (https://github.com/vyasr)
  - Dante Gama Dessavre (https://github.com/dantegd)
  - Ray Douglass (https://github.com/raydouglass)

URL: https://github.com/rapidsai/raft/pull/2296
---
 dependencies.yaml               | 2 +-
 python/raft-dask/pyproject.toml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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`.