9df8cd324207d3924439c3d912229811e34687f0,python/dgl/model_zoo/chem/jtnn/jtnn_vae.py,DGLJTNNVAE,__init__,#DGLJTNNVAE#Any#Any#Any#Any#Any#,34

Before Change

            if vocab_file is None:
                vocab_file = "{}/jtnn/{}.txt".format(
                    get_download_dir(), "vocab")
                self.vocab = Vocab([x.strip("\r\n ")
                                    for x in open(vocab_file)])
            else:
                self.vocab = Vocab([x.strip("\r\n ")
                                    for x in open(vocab_file)])
        else:

After Change

    def __init__(self, hidden_size, latent_size, depth, vocab=None, vocab_file=None):
        super(DGLJTNNVAE, self).__init__()
        if vocab is None:
            if vocab_file is None:
                vocab_file = "{}/jtnn/{}.txt".format(
                    get_download_dir(), "vocab")

            self.vocab = Vocab([x.strip("\r\n ")
                                for x in open(vocab_file)])
        else:
            self.vocab = vocab

Instances