the pip package is called 'torch-geometric' not 'pytorch_geometric'

rapidsai · Oct 21, 2024 · fad3e8b · fad3e8b
1 parent 4033210
commit fad3e8b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -151,6 +151,7 @@ files:
       table: project.optional-dependencies
       key: test
     includes:
+      - depends_on_pyg
       - depends_on_pylibwholegraph
       - depends_on_pytorch
       - test_python_common
@@ -364,7 +365,6 @@ dependencies:
       - output_types: [conda, pyproject, requirements]
         packages:
           - ogb
-          - torch-geometric>=2.5,<2.6
   test_python_pylibwholegraph:
     common:
       - output_types: [conda, pyproject, requirements]
@@ -434,9 +434,12 @@ dependencies:
 
   depends_on_pyg:
     common:
-      - output_types: [conda, pyproject, requirements]
+      - output_types: [conda]
         packages:
           - pytorch_geometric>=2.5,<2.6
+      - output_types: [pyproject, requirements]
+        packages:
+          - torch-geometric>=2.5,<2.6
 
   depends_on_pylibwholegraph:
     common:

diff --git a/python/cugraph-pyg/pyproject.toml b/python/cugraph-pyg/pyproject.toml
@@ -34,7 +34,7 @@ dependencies = [
     "numpy>=1.23,<3.0a0",
     "pandas",
     "pylibcugraphops==24.12.*,>=0.0.0a0",
-    "pytorch_geometric>=2.5,<2.6",
+    "torch-geometric>=2.5,<2.6",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]