fixed failing unit test

mbatchkarov · Nov 10, 2015 · bf3a996 · bf3a996
1 parent f354efc
commit bf3a996
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/eval/pipeline/feature_extractors.py b/eval/pipeline/feature_extractors.py
@@ -8,11 +8,13 @@ class FeatureExtractor(object):
     def __init__(self, remove_features_with_NER=False,
                  extract_unigram_features='J,N,V',
                  extract_phrase_features=['AN', 'NN', 'SVO'],
-                 standard_ngram_features=0):
+                 standard_ngram_features=0,
+                 remove_pos=False):
         self.remove_features_with_NER = remove_features_with_NER
         self.extract_phrase_features = extract_phrase_features
         self.extract_unigram_features = extract_unigram_features
         self.standard_ngram_features = standard_ngram_features
+        self.remove_pos = remove_pos
 
     def update(self, **kwargs):
         self.__dict__.update(**kwargs)