alignn.py, line 294, g, lg, lat = g

[02pdbq20]

Error message：
Traceback (most recent call last):
File "c:\Users\12732\Desktop\python\ml\alignn-2024.10.30\temp\predicate.py", line 27, in
g, lg ,lat = Graph.atom_dgl_multigraph(
edata_schemes={'h': Scheme(shape=(), dtype=torch.float32)}))
Traceback (most recent call last):
File "c:\Users\12732\Desktop\python\ml\alignn-2024.10.30\temp\predicate.py", line 31, in
model([g.to(device), lg.to(device)])
File "C:\xuexi\anaconda3\envs\sdsd_torch\lib\site-packages\torch\nn\modules\module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "C:\xuexi\anaconda3\envs\sdsd_torch\lib\site-packages\alignn\models\alignn.py", line 294, in forward
g, lg, lat = g
ValueError: not enough values to unpack (expected 3, got 2)

modified：
File "C:\xuexi\anaconda3\envs\sdsd_torch\lib\site-packages\alignn\models\alignn.py", line 294, in forward
g, lg = g

Can operate normally

[02pdbq20]

g, lg, *_ = g might be a better choice