wengong-jin / icml18-jtnn

Junction Tree Variational Autoencoder for Molecular Graph Generation (ICML 2018)
MIT License
509 stars 190 forks source link

AttributeError: 'MolTreeNode' object has no attribute 'cid' #35

Closed lihan1997 closed 5 years ago

lihan1997 commented 5 years ago

Hi! This error occurs when I ran sample.py in fast_molvae without any changes.

Traceback (most recent call last): File "/fast_molvae/sample.py", line 39, in print model.sample_prior() File "/jtnn_vae.py", line 119, in sample_prior return self.decode(z_tree, z_mol, prob_decode) File "/fast_jtnn/jtnn_vae.py", line 200, in decode jtenc_holder, mess_dict = JTNNEncoder.tensorize_nodes(pred_nodes, scope) File "/fast_jtnn/jtnn_enc.py", line 73, in tensorize_nodes cid = y.cid[1] if y.cid[0] == x.idx else 0 AttributeError: 'MolTreeNode' object has no attribute 'cid'

Process finished with exit code 1

Maybe something wrong with your codes?

Thank you!

wengong-jin commented 5 years ago

Hi,

The issue is fixed. I accidentally committed some codes that are still developing :)

My apologies

lihan1997 commented 5 years ago

Thanks!