From 72358aa38df9b073482c1bca9a2eb56539407d37 Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Tue, 12 Nov 2024 16:54:01 -0600
Subject: [PATCH] move cugraph-dgl, cugraph-pyg, wholegraph to cugraph-gnn repo

---
 rapids-metadata.json            | 32 ++++++++++++++++----------------
 src/rapids_metadata/__init__.py | 12 ++++++++++++
 2 files changed, 28 insertions(+), 16 deletions(-)

diff --git a/rapids-metadata.json b/rapids-metadata.json
index 97cc4dc..12a2dea 100644
--- a/rapids-metadata.json
+++ b/rapids-metadata.json
@@ -580,18 +580,10 @@
               "has_cuda_suffix": true,
               "publishes_prereleases": true
             },
-            "cugraph-dgl": {
-              "has_cuda_suffix": true,
-              "publishes_prereleases": true
-            },
             "cugraph-equivariant": {
               "has_cuda_suffix": true,
               "publishes_prereleases": true
             },
-            "cugraph-pyg": {
-              "has_cuda_suffix": true,
-              "publishes_prereleases": true
-            },
             "cugraph-service-client": {
               "has_cuda_suffix": false,
               "publishes_prereleases": true
@@ -610,6 +602,22 @@
             }
           }
         },
+        "cugraph-gnn": {
+          "packages": {
+            "cugraph-dgl": {
+              "has_cuda_suffix": true,
+              "publishes_prereleases": true
+            },
+            "cugraph-pyg": {
+              "has_cuda_suffix": true,
+              "publishes_prereleases": true
+            },
+            "pylibwholegraph": {
+              "has_cuda_suffix": true,
+              "publishes_prereleases": true
+            }
+          }
+        },
         "cugraph-ops": {
           "packages": {
             "libcugraphops": {
@@ -789,14 +797,6 @@
               "publishes_prereleases": true
             }
           }
-        },
-        "wholegraph": {
-          "packages": {
-            "pylibwholegraph": {
-              "has_cuda_suffix": true,
-              "publishes_prereleases": true
-            }
-          }
         }
       }
     }
diff --git a/src/rapids_metadata/__init__.py b/src/rapids_metadata/__init__.py
index 2253aa6..b23b0ad 100644
--- a/src/rapids_metadata/__init__.py
+++ b/src/rapids_metadata/__init__.py
@@ -167,3 +167,15 @@
 )
 
 all_metadata.versions["24.12"] = deepcopy(all_metadata.versions["24.10"])
+
+del all_metadata.versions["24.12"].repositories["cugraph"].packages["cugraph-dgl"]
+del all_metadata.versions["24.12"].repositories["cugraph"].packages["cugraph-pyg"]
+del all_metadata.versions["24.12"].repositories["wholegraph"]
+
+all_metadata.versions["24.12"].repositories["cugraph-gnn"] = RAPIDSRepository(
+    packages={
+        "cugraph-dgl": RAPIDSPackage(),
+        "cugraph-pyg": RAPIDSPackage(),
+        "pylibwholegraph": RAPIDSPackage(),
+    }
+)