bug fix

README.md

@@ -78,11 +78,9 @@ class TransformersUD(object):
         m[i+1,0 if i==j else j+1]=s[i][b[i][j]]+e[i][b[i][j+1]-1]
     h=ufal.chu_liu_edmonds.chu_liu_edmonds(m)[0]
     if [0 for i in h if i==0]!=[0]:
-      try:
-        i=[p for s,e,p in w].index("root")+1
-      except:
-        i=numpy.nanargmax(m[:,0])
-      m[0:i,0]=m[i+1:,0]=numpy.nan
+      i=([p for s,e,p in w]+["root"]).index("root")
+      j=numpy.nanargmax(m[:,0]) if i==n else i+1
+      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"
     for i,(s,e,p) in enumerate(w,1):