feat: add support mutary relation

Author: wajdikhattel

treedlib/features.py

@@ -1,7 +1,7 @@
 from treedlib.templates import *
 import lxml.etree as et
 
-def compile_relation_feature_generator(dictionaries=None, opts={}):
+def compile_relation_feature_generator(dictionaries=None, opts={}, is_multary=False):
   """
   Given optional arguments, returns a generator function which accepts an xml root
   and two lists of mention indexes, and will generate relation features for this relation
@@ -66,7 +66,9 @@ def compile_relation_feature_generator(dictionaries=None, opts={}):
       templates.append(DictionaryIntersect(SeqBetween(), d_name, d))
 
   # return generator function
-  return Compile(templates).apply_relation
+    if is_multary:
+      return Compile(templates).apply_multary_relation
+    return Compile(templates).apply_relation
 
 """
 For calibrating the bin sizes

treedlib/templates.py

@@ -428,5 +428,8 @@ def apply_mention(self, root, mention_idxs, dict_sub={}, stopwords=None):
   def apply_relation(self, root, mention1_idxs, mention2_idxs, dict_sub={}, stopwords=None):
     return self.apply(root, [mention1_idxs, mention2_idxs], dict_sub=dict_sub, stopwords=stopwords)
 
+  def apply_multary_relation(self, root, mentions, dict_sub={}, stopwords=None):
+    return self.apply(root, mentions, dict_sub=dict_sub, stopwords=stopwords)
+  
   def __repr__(self):
     return '\n'.join(str(op) for op in self._iterops())