Refactor method name to be more clear

PetrochukM · Apr 26, 2018 · 5589edf · 5589edf
1 parent a28b430
commit 5589edf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchnlp/word_to_vector/pretrained_word_vectors.py b/torchnlp/word_to_vector/pretrained_word_vectors.py
@@ -74,7 +74,7 @@ def __init__(self,
     def __contains__(self, token):
         return token in self.stoi
 
-    def _get_token(self, token):
+    def _get_token_vector(self, token):
         """Return embedding for token or for UNK if token not in vocabulary"""
         if token in self.stoi:
             return self.vectors[self.stoi[token]]