051bd416d3c41002f6d58b9dd71516a27243d178,thumt/launcher/translator.py,,main,#Any#,137

Before Change

                if idx >= 0:
                    output = decoded[:idx].strip()
                else:
                    output = decoded.strip()

                outfile.write("%s\n" % output)

After Change

        with open(params.output, "w") as outfile:
            for output in restored_outputs:
                decoded = []
                for idx in output:
                    if idx == params.mapping["target"][params.eos]:
                        break
                    decoded.append(vocab[idx])

                decoded = " ".join(decoded)
                outfile.write("%s\n" % decoded)

Instances