Missing ``class_resolver`` dependency

😵 Describe the installation problem

Recently I’m receiving the following error caused by loading a GNN layer object from PyG due to a missing dependency on class_resolver.

  File "/opt/hostedtoolcache/Python/3.8.12/x64/lib/python3.8/site-packages/torch_geometric/nn/models/basic_gnn.py", line 60, in __init__
    from class_resolver.contrib.torch import activation_resolver
ModuleNotFoundError: No module named 'class_resolver'

The issue is resolved by simply installing this (pip install class_resolver). I also notice that this requirement is included in the “full” install requirements and will be installed automatically if I do pip install torch-geometric [full]. However, I’m using the latest PyG on github in my github actions pip install git+https://github.com/pyg-team/pytorch_geometric.git. I’m wondering would it be possible to pass the [full] option to pip in this case? I tried pip install git+https://github.com/pyg-team/pytorch_geometric.git [full] and it did not work:

ERROR: Invalid requirement: '[full]'

Environment

• PyG version: latest (pip install git+https://github.com/pyg-team/pytorch_geometric)
• PyTorch version: 1.9.0
• OS: ubuntu-latest
• Python version: 3.8
• CUDA/cuDNN version: CPU
• How you installed PyTorch and PyG (conda, pip, source): pip
• Any other relevant information (e.g., version of torch-scatter):

[rusty1s]

Currently, class_resolver is used inside MLP and basic GNN models for resolving activation functions from PyTorch. We could consider integrating this functionality directly inside PyG to avoid the extra dependency. Otherwise, falling back to pip install class-resolver is the only workaround I am aware of.

Pinging @cthoyt here as well.

[rusty1s]

This should work:

pip install git+https://github.com/pyg-team/pytorch_geometric.git#egg=torch-geometric[full]

Alternatively, you can just install class-resolver after installing PyG:

pip install "git+https://github.com/pyg-team/pytorch_geometric.git
pip install class-resolver