6a5916968838686e332fb68a10cb30222843fbd9,torchnlp/word_to_vector/pretrained_word_vectors.py,_PretrainedWordVectors,__getitem__,#_PretrainedWordVectors#Any#,77

Before Change

        return token in self.stoi

    def __getitem__(self, token):
        if token in self.stoi:
            return self.vectors[self.stoi[token]]
        else:
            return self.unk_init(torch.Tensor(self.dim))

    def __len__(self):
        return len(self.vectors)

    def __str__(self):

After Change

        if isinstance(tokens, list) or isinstance(tokens, tuple):
            vector_list = [self._get_token(token) for token in tokens]
            return torch.stack(vector_list)
        elif isinstance(tokens, str):
            token = tokens
            return self._get_token(token)
        else:
            raise TypeError(""__getitem__" method can only be used with types"
                            ""str", "list", or "tuple" as parameter")

Instances