From 68fa626eecd6cbd300db00db150dad6fdf6f278c Mon Sep 17 00:00:00 2001
From: Vyas Ramasubramani <vyasr@nvidia.com>
Date: Wed, 15 Nov 2023 18:05:25 -0800
Subject: [PATCH] Make dependency lists compliant

---
 dependencies.yaml | 16 +++++++++-------
 pyproject.toml    | 13 +++++--------
 2 files changed, 14 insertions(+), 15 deletions(-)

diff --git a/dependencies.yaml b/dependencies.yaml
index 57ac87ee7..30e1f75cc 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -138,21 +138,23 @@ dependencies:
           - numpy>=1.21
           - pandas>=1.3,<1.6.0dev0
           - pynvml>=11.0.0,<11.5
-          - rapids-dask-dependency=24.02.*
+          - rapids-dask-dependency==24.02.*
           - zict>=2.0.0
   test_python:
     common:
       - output_types: [conda, requirements, pyproject]
         packages:
-          - cudf=24.02
-          - dask-cudf=24.02
-          - kvikio=24.02
+          - cudf==24.02.*
+          - dask-cudf==24.02.*
+          - kvikio==24.02.*
           - pytest
           - pytest-cov
+          - ucx-py==0.36.*
+      - output_types: [conda]
+        packages:
+          - distributed-ucxx==0.35.*
           - ucx-proc=*=gpu
-          - ucx-py=0.36
-          - ucxx=0.35
-          - distributed-ucxx=0.35
+          - ucxx==0.35.*
     specific:
       - output_types: conda
         matrices:
diff --git a/pyproject.toml b/pyproject.toml
index 2002b1874..da065b80e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,7 +20,7 @@ dependencies = [
     "numpy>=1.21",
     "pandas>=1.3,<1.6.0dev0",
     "pynvml>=11.0.0,<11.5",
-    "rapids-dask-dependency=24.02.*",
+    "rapids-dask-dependency==24.02.*",
     "zict>=2.0.0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
 classifiers = [
@@ -48,15 +48,12 @@ docs = [
     "sphinx-rtd-theme>=0.5.1",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
 test = [
-    "cudf=24.02",
-    "dask-cudf=24.02",
-    "distributed-ucxx=0.35",
-    "kvikio=24.02",
+    "cudf==24.02.*",
+    "dask-cudf==24.02.*",
+    "kvikio==24.02.*",
     "pytest",
     "pytest-cov",
-    "ucx-proc=*=gpu",
-    "ucx-py=0.36",
-    "ucxx=0.35",
+    "ucx-py==0.36.*",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]