move cugraph-dgl, cugraph-pyg, wholegraph to cugraph-gnn repo

rapidsai · Nov 12, 2024 · 72358aa · 72358aa
1 parent 42b59b7
commit 72358aa
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
@@ -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
@@ -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(),
+    }
+)