def decode(x, translation):
enc = ["".join([str(int(y)) for y in e[1:]]) for e in x if e[0] == 1]
fs = [Chem.Mol(translation[e]) for e in enc]
if not fs:
return Chem.Mol()
return join_fragments(fs)
def decode_molecule(enc, decodings):
fs = [Chem.Mol(decodings[x]) for x in enc.split("-")]
return join_fragments(fs)