update handling of GNN dependencies

[jameslamb]

Follow up to #4935

There, in #4935 (comment), we discussed the that GNN packages shouldn't need to be installed in docs builds any more, as no docs in this repo (including notebooks) require them.

This PR limits dependencies on the GNN packages to only the places they're needed.

libwholegraph / pylibwholegraph

git grep -i -E 'wholegraph'

Optional runtime dependency of cugraph:

cugraph/python/cugraph/cugraph/gnn/feature_storage/feat_storage.py

Line 23 in 2873ff9

wgth = import_optional("pylibwholegraph.torch")

And optional test-time dependency of cugraph:

cugraph/python/cugraph/cugraph/tests/data_store/test_gnn_feat_storage_wholegraph.py

Lines 24 to 25 in 2873ff9

	pylibwholegraph = import_optional("pylibwholegraph")
	wmb = import_optional("pylibwholegraph.binding.wholememory_binding")

But not used in any docs.

Changes: Removed from docs environment.

cugraph-dgl

git grep -i -E 'cugraph.*dgl'

Not used anywhere in this repo.

Changes: Removed all remaining references.

cugraph-pyg

git grep -i -E 'cugraph.*pyg'

Only used as an optional import in cugraph-service-client's tests (which are not run in CI for wheels).

cugraph/python/cugraph-service/tests/test_remote_graph.py

Line 662 in 2873ff9

from cugraph_pyg.loader.dispatch import call_cugraph_algorithm

Changes: Removed from the docs environment, added a [test] extra to cugraph-service-client including this.

Notes for Reviewers

Related to these issues about moving more GNN stuff out of this repo:

• [FEA] Migrate All PyTorch-Dependent Code to cuGraph-GNN #4822
• Investigate Migrating cugraph.gnn Subpackage #4407

[jameslamb]

/ok to test

[bdice]

We should ask the graph team to purge that reference to cugraph_pyg in a follow-up. It has a "FIXME" and it introduces an awkward circular dependency between repositories.

[jameslamb]

should ask the graph team to purge that reference to cugraph_pyg in a follow-up. It has a "FIXME" and it introduces an awkward circular dependency between repositories.

Agreed! cc @alexbarghi-nv

[jameslamb]

/merge