Fail to download ogb-molpcba dataset caused by the version of torch_geometric.

[CBQu]

I ran python wilds/wilds/download_datasets.py --root_dir data --datasets ogb-molpcba.
And got an error message like this.

Traceback (most recent call last):
  File "wilds/wilds/download_datasets.py", line 34, in <module>
    main()
  File "wilds/wilds/download_datasets.py", line 27, in main
    wilds.get_dataset(
  File "..../wilds/get_dataset.py", line 52, in get_dataset
    from wilds.datasets.ogbmolpcba_dataset import OGBPCBADataset
  File "..../wilds/datasets/ogbmolpcba_dataset.py", line 7, in <module>
    from torch_geometric.data.dataloader import Collater as PyGCollater
ModuleNotFoundError: No module named 'torch_geometric.data.dataloader'

I found it is caused by torch_geometric changing the module name or moving the function.
I fixed it by changing from torch_geometric.data.dataloader import Collater as PyGCollater to from torch_geometric.loader.dataloader import Collater as PyGCollater. And successfully downloaded the data.

I guess you could check the dependency and solve it. The version of my torch_geometric is 2.0.2.

BTW This benchmark is very useful. It would be nice to have a TensorFlow version. I am looking forward to it.

[kohpangwei]

Thank you for flagging this! We've made the fix at 3da4b35.