it seems a bug in your reinforcement.py line 114

[tbwxmu]

Hi all,
A little bit confuse about the deaded while loop, are you want to skip the invalid SMILES?
So add "break" next to "reward = 0" can work. However, I think you want to penalize the invalid smiles, so just replace reward to some negative value?