Remove changes to __contains__() method

tests/word_to_vector/test_fast_text.py

@@ -46,10 +46,6 @@ def test_fasttext_list_arguments(mock_urlretrieve):
     list(vectors[['the', 'of']].shape) == [2, 300]
     list(vectors[('the', 'of')].shape) == [2, 300]
 
-    # Test implementation of __contains()__ for token list and tuple
-    assert ['the', 'of', 'a'] in vectors == [True, True, False]
-    assert ('the', 'of', 'a') in vectors == [True, True, False]
-
     # Clean up
     os.remove(os.path.join(directory, 'wiki.simple.vec.pt'))
 

torchnlp/word_to_vector/pretrained_word_vectors.py

@@ -71,15 +71,8 @@ def __init__(self,
         self.name = name
         self.cache(name, cache, url=url)
 
-    def __contains__(self, tokens):
-        if isinstance(tokens, list) or isinstance(tokens, tuple):
-            return [token in self.stoi for token in tokens]
-        elif isinstance(tokens, str):
-            token = tokens
-            return token in self.stoi
-        else:
-            raise TypeError("'__contains__' method can only be used with types"
-                            "'str', 'list', or 'tuple' as parameter")
+    def __contains__(self, token):
+        return token in self.stoi
 
     def _get_token(self, token):
         """Return embedding for token or for UNK if token not in vocabulary"""