bug fix

README.md

@@ -51,7 +51,7 @@ class UDgoeswith(object):
       m[i,0],m[i,i],p[i,0]=m[i,i],numpy.nan,p[i,i]
     h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     if [0 for i in h if i==0]!=[0]:
-      m[:,0]+=numpy.where(m[:,0]<numpy.nanmax(m[:,0]),numpy.nan,0)
+      m[:,0]+=numpy.where(m[:,0]==numpy.nanmax(m[[i for i,j in enumerate(h) if j==0],0]),0,numpy.nan)
       m[[i for i,j in enumerate(h) if j==0]]+=[0 if i==0 or j==0 else numpy.nan for i,j in enumerate(h)]
       h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     u="# text = "+text+"\n"

ud.py

@@ -27,7 +27,7 @@ class UniversalDependenciesPipeline(TokenClassificationPipeline):
       m[i,0],m[i,i],p[i,0]=m[i,i],numpy.nan,p[i,i]
     h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     if [0 for i in h if i==0]!=[0]:
-      m[:,0]+=numpy.where(m[:,0]<numpy.nanmax(m[:,0]),numpy.nan,0)
+      m[:,0]+=numpy.where(m[:,0]==numpy.nanmax(m[[i for i,j in enumerate(h) if j==0],0]),0,numpy.nan)
       m[[i for i,j in enumerate(h) if j==0]]+=[0 if i==0 or j==0 else numpy.nan for i,j in enumerate(h)]
       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]