From 8556f48c489c4ced33f320711ddafd258ef7f757 Mon Sep 17 00:00:00 2001
From: divyegala <divyegala@gmail.com>
Date: Tue, 7 May 2024 14:52:07 -0700
Subject: [PATCH] libucx run dependency

---
 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 a336aa1577..6ed26a3b14 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -100,6 +100,7 @@ files:
       table: project
     includes:
       - run_raft_dask
+      - depends_on_ucx_run
   py_test_raft_dask:
     output: pyproject
     pyproject_dir: python/raft-dask
@@ -109,7 +110,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 0181bef4ce..4718d51a35 100644
--- a/python/raft-dask/pyproject.toml
+++ b/python/raft-dask/pyproject.toml
@@ -36,6 +36,7 @@ requires-python = ">=3.9"
 dependencies = [
     "dask-cuda==24.6.*",
     "joblib>=0.11",
+    "libucx>=1.15.0",
     "numba>=0.57",
     "numpy>=1.23,<2.0a0",
     "pylibraft==24.6.*",
@@ -53,7 +54,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`.