add post filters to custom spacy extractor

usc-isi-i2 · Jan 23, 2018 · 2274ff4 · 2274ff4
1 parent 2389a33
commit 2274ff4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/etk/core.py b/etk/core.py
@@ -1527,6 +1527,10 @@ def extract_using_custom_spacy(self, d, config, field_rules=None):
         results = self._relevant_text_from_context(d[_SIMPLE_TOKENS_ORIGINAL_CASE],
                                                    custom_spacy_extractor.extract(field_rules, nlp_doc, self.nlp),
                                                    config[_FIELD_NAME])
+        if _POST_FILTER in config:
+            post_filters = config[_POST_FILTER]
+            results = self.run_post_filters_results(results, post_filters)
+
         return results
 
     def infer_rule_using_custom_spacy(self, d, positive_eg):