Merge pull request #111 from writerai/NLP-82--add-tag-to-relative-x-is-y

Add tag to relative_x_is_y

replacy/default_match_hooks.py

@@ -214,8 +214,8 @@ def relative_x_is_y(
             "children_or_ancestors must be set to either `children` or `ancestors`"
         )
 
-    if pos_or_dep not in ["pos", "dep"]:
-        raise ValueError("pos_or_dep must be set to either `pos` or `dep`!")
+    if pos_or_dep not in ["pos", "dep", "tag"]:
+        raise ValueError("pos_or_dep must be set to either `pos`, `dep`, or `tag`!")
 
     def _in_children(doc, start, end):
         if end >= len(doc):
@@ -227,9 +227,9 @@ def _in_children(doc, start, end):
                     [child.pos_ == val for tok in match_span for child in tok.children]
                 )
             elif pos_or_dep == "dep":
-                return any(
-                    [child.dep_ == val for tok in match_span for child in tok.children]
-                )
+                return any([child.dep_ == val for tok in match_span for child in tok.children])
+            elif pos_or_dep == "tag":
+                return any([child.tag_ == val for tok in match_span for child in tok.children])
 
     def _in_ancestors(doc, start, end):
         if end >= len(doc):
@@ -248,6 +248,12 @@ def _in_ancestors(doc, start, end):
                     if ancestor and ancestor.dep_ == val:
                         return True
                 return False
+            if pos_or_dep == "tag":
+                for t in match_span:
+                    ancestor = list(t.ancestors)[0] if len(list(t.ancestors)) else None
+                    if ancestor and ancestor.tag_ == val:
+                        return True
+                return False
 
     if children_or_ancestors == "children":
         return _in_children