diff --git a/dependencies.yaml b/dependencies.yaml index 4b2ad1d..748a9d2 100644 --- 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 index 3a1d5bf..ba13b8f 100644 --- 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]