chang from & to in to fix this issue.

duyphongdn1997 · Mar 22, 2018 · 6f77bd3 · 6f77bd3
1 parent 39701cb
commit 6f77bd3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/drqa/tokenizers/spacy_tokenizer.py b/drqa/tokenizers/spacy_tokenizer.py
@@ -27,7 +27,7 @@ def __init__(self, **kwargs):
         nlp_kwargs = {'parser': False}
         if not {'lemma', 'pos', 'ner'} in self.annotators:
             nlp_kwargs['tagger'] = False
-        if not {'ner'} & self.annotators:
+        if not {'ner'} in self.annotators:
             nlp_kwargs['entity'] = False
         self.nlp = spacy.load(model, **nlp_kwargs)
 
@@ -37,7 +37,7 @@ def tokenize(self, text):
         tokens = self.nlp.tokenizer(clean_text)
         if {'lemma', 'pos', 'ner'} in self.annotators:
             self.nlp.tagger(tokens)
-        if {'ner'} & self.annotators:
+        if {'ner'} in self.annotators:
             self.nlp.entity(tokens)
 
         data = []