Running colab exmaple got an error ModuleNotFoundError: No module named 'torch_geometric'

[pengzhangzhi]

Running the colab notebook 1.7.0 | Creating Datasets from the PDB and got ModuleNotFoundError: No module named 'torch_geometric'

ModuleNotFoundError                       Traceback (most recent call last)
[<ipython-input-3-a0a0d608f1af>](https://localhost:8080/#) in <cell line: 3>()
      1 get_ipython().system('pip install graphein')
      2 from rich import inspect
----> 3 from graphein.ml.datasets import PDBManager
      4 
      5 manager = PDBManager(root_dir=".")

1 frames
[/usr/local/lib/python3.10/dist-packages/graphein/ml/conversion.py](https://localhost:8080/#) in <module>
     15 import torch
     16 from loguru import logger as log
---> 17 from torch_geometric.utils.undirected import to_undirected
     18 
     19 from graphein.utils.dependencies import import_message

ModuleNotFoundError: No module named 'torch_geometric'

[a-r-j]

Thanks for flagging this! You can follow the docs here: https://pytorch-geometric.readthedocs.io/en/2.1.0/notes/installation.html

[pengzhangzhi]

Yes! I just want to raise the issue here so that other users can run your example.