bug fix

README.md

@@ -79,7 +79,7 @@ class TransformersUD(object):
     h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     if [0 for i in h if i==0]!=[0]:
       i=([p for s,e,p in w]+["root"]).index("root")
-      j=numpy.nanargmax(m[:,0]) if i==n else i+1
+      j=i+1 if i<n else numpy.nanargmax(m[:,0])
       m[0:j,0]=m[j+1:,0]=numpy.nan
       h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     u="# text = "+text.replace("\n"," ")+"\n"