bug fix

ud.py

@@ -1,15 +1,15 @@
 from transformers import TokenClassificationPipeline
 
 class UniversalDependenciesPipeline(TokenClassificationPipeline):
-  def _forward(self,model_input):
+  def _forward(self,model_inputs):
     import torch
-    v=model_input["input_ids"][0].tolist()
+    v=model_inputs["input_ids"][0].tolist()
     with torch.no_grad():
       e=self.model(input_ids=torch.tensor([v[0:i]+[self.tokenizer.mask_token_id]+v[i+1:]+[j] for i,j in enumerate(v[1:-1],1)]))
-    return {"logits":e.logits[:,1:-2,:],**model_input}
-  def postprocess(self,model_output,**kwargs):
+    return {"logits":e.logits[:,1:-2,:],**model_inputs}
+  def postprocess(self,model_outputs,**kwargs):
     import numpy
-    e=model_output["logits"].numpy()
+    e=model_outputs["logits"].numpy()
     r=[1 if i==0 else -1 if j.endswith("|root") else 0 for i,j in sorted(self.model.config.id2label.items())]
     e+=numpy.where(numpy.add.outer(numpy.identity(e.shape[0]),r)==0,0,numpy.nan)
     g=self.model.config.label2id["X|_|goeswith"]
@@ -25,7 +25,7 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
       k,h=z[numpy.nanargmax(m[z,z])],numpy.nanmin(m)-numpy.nanmax(m)
       m[:,z]+=[[0 if j in z and (i!=j or i==k) else h for i in z] for j in range(m.shape[0])]
       h=self.chu_liu_edmonds(m)
-    v=[(s,e) for s,e in model_output["offset_mapping"][0].tolist() if s<e]
+    v=[(s,e) for s,e in model_outputs["offset_mapping"][0].tolist() if s<e]
     q=[self.model.config.id2label[p[j,i]].split("|") for i,j in enumerate(h)]
     g="aggregation_strategy" in kwargs and kwargs["aggregation_strategy"]!="none"
     if g:
@@ -34,7 +34,7 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
           h=[b if i>b else b-1 for a,b in enumerate(h) if i!=a]
           v[i-1]=(v[i-1][0],v.pop(i)[1])
           q.pop(i)
-    t=model_output["sentence"].replace("\n"," ")
+    t=model_outputs["sentence"].replace("\n"," ")
     u="# text = "+t+"\n"
     for i,(s,e) in enumerate(v):
       u+="\t".join([str(i+1),t[s:e],t[s:e] if g else "_",q[i][0],"_","|".join(q[i][1:-1]),str(0 if h[i]==i else h[i]+1),q[i][-1],"_","_" if i+1<len(v) and e<v[i+1][0] else "SpaceAfter=No"])+"\n"