fixed code and updated version

Author: anuragkumar95

README.md

@@ -1,6 +1,6 @@
 # Python JSON-NLP Module
 
-(C) 2019 by [Damir Cavar], [Oren Baldinger], [Maanvitha Gongalla], Anurag Kumar, Murali Kammili
+(C) 2019 by [Damir Cavar], [Oren Baldinger], [Maanvitha Gongalla], [Anurag Kumar], Murali Kammili
 
 Brought to you by the [NLP-Lab.org]!
 
@@ -112,6 +112,7 @@ Other parameters specific to your pipeline implementation can be passed as well:
 
 [Damir Cavar]: http://damir.cavar.me/ "Damir Cavar"
 [Oren Baldinger]: https://oren.baldinger.me/ "Oren Baldinger"
+[Anurag Kumar]: https://github.com/anuragkumar95/ "Anurag Kumar"
 [Maanvitha Gongalla]: https://maanvithag.github.io/MaanvithaGongalla/
 [NLP-Lab.org]: http://nlp-lab.org/ "NLP-Lab.org"
 [JSON-NLP]: https://github.com/dcavar/JSON-NLP "JSON-NLP"

pyjsonnlp/__init__.py

@@ -14,7 +14,7 @@
 from typing import List
 
 name = "pyjsonnlp"
-__version__ = "0.2.30"
+__version__ = "0.2.33"
 
 
 def get_base() -> OrderedDict:
@@ -113,7 +113,7 @@ def find_head(doc: OrderedDict, token_ids: List[int], sentence_id: int, style='u
     #print(sentence_id, doc['dependencies'],doc['dependencies']['trees'])
     if len(token_ids) == 0:
         return None
-    arcs = doc['dependencies']['trees'][sentence_id-1]
+    arcs = doc['dependencies'][sentence_id-1]['trees']
     govs = set(token_ids)
     for x in arcs:
         if x["dep"] in govs and x["gov"] in govs:

setup.py

@@ -6,7 +6,7 @@
 
 setuptools.setup(
     name="pyjsonnlp",
-    version='0.2.30',
+    version='0.2.33',
     python_requires='>=3.6',
     author="Damir Cavar, Oren Baldinger, Maanvitha Gongalla, Anurag Kumar, Murali Kammili",
     author_email="damir@cavar.me",