Giters

cvignac / MiDi

MiDi: Mixed Graph and 3D Denoising Diffusion for Molecule Generation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ImportError: cannot import name 'LightningDataset' from 'torch_geometric.data'

zhi520 opened this issue · comments

commented

MiDi/midi_src/datasets/adaptive_loader.py", line 6, in
from torch_geometric.data import LightningDataset

ImportError: cannot import name 'LightningDataset' from 'torch_geometric.data'

commented

It's probably related to your pytorch_geometric version, can you check it using pip list?

Clement

commented

Hi, I found that the LightningDataset moved to torch_geometric.data.lightning.LightningDataset. Maybe you could try this.

commented

Thank you, I have fixed the code accordingly.