move cugraph-dgl, cugraph-pyg, wholegraph to cugraph-gnn repo (#31)

rapidsai · Nov 15, 2024 · 4f8bb33 · 4f8bb33
1 parent e8bbce7
commit 4f8bb33
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 16 deletions.
diff --git 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
@@ -606,6 +598,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": {
@@ -793,14 +801,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
@@ -168,7 +168,19 @@
 
 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["cugraph"].packages["nx-cugraph"]
+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(),
+    }
+)
+
 all_metadata.versions["24.12"].repositories["nx-cugraph"] = RAPIDSRepository(
     packages={
         "nx-cugraph": RAPIDSPackage(),