aggregation_strategy

ud.py

@@ -25,7 +25,7 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
       h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     v=[(s,e) for s,e in model_output["offset_mapping"][0].tolist() if s<e]
     q=[self.model.config.id2label[p[i,j]].split("|") for i,j in enumerate(h)]
-    if "aggregation_strategy" in kwargs and kwargs["aggregation_strategy"]=="simple":
+    if "aggregation_strategy" in kwargs and kwargs["aggregation_strategy"]!="none":
       for i,j in reversed(list(enumerate(q[2:],2))):
         if j[-1]=="goeswith":
           h=[b if i>b else b-1 for a,b in enumerate(h) if i!=a]