Fix augment_sentence (thanks @xiangc2). Closes amazon-science#4.

giove91 · giove91 · commit c6110a9e79cd · 2021-05-25T15:22:34.000-07:00
diff --git a/utils.py b/utils.py
@@ -100,7 +100,7 @@ def augment_sentence(tokens: List[str], augmentations: List[Tuple[List[tuple], i
 
     # check that the entities have a tree structure (if two entities overlap, then one is contained in
     # the other), and build the entity tree
-    root = (None, 0, len(tokens))   # this node represents the entire sentence
+    root = (None, -1, len(tokens))  # this node represents the entire sentence
     entity_tree = {root[1:]: []}    # list of children of each node
     current_stack = [root]          # where we are in the tree
 
