is this a type error in jtnn_vae?

[asawanggaa]

line 91, self.stereo seems to be a bool type
if self.stereo:
stereo_loss, stereo_acc = self.stereo(mol_batch, mol_vec)

which self.stereo init here, line 46
self.stereo = stereo

[Wengong Jin]

Thanks for pointing this out. The issue has been fixed.