Merge pull request #341 from astraszab/hyponym-detector-index-error-fix

Hyponym detector index error fix
allenai · Mar 19, 2021 · 4ade4ec · 4ade4ec
2 parents 5df54e4 + d191052
commit 4ade4ec
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/scispacy/hyponym_detector.py b/scispacy/hyponym_detector.py
@@ -70,16 +70,20 @@ def expand_to_noun_compound(self, token: Token, doc: Doc):
 
         start = token.i
         while True:
-            previous = doc[start - 1]
-            if previous.pos_ in {"PROPN", "NOUN", "PRON"}:
+            if start - 1 < 0:
+                break
+            previous_token = doc[start - 1]
+            if previous_token.pos_ in {"PROPN", "NOUN", "PRON"}:
                 start -= 1
             else:
                 break
 
         end = token.i + 1
         while True:
-            previous = doc[end]
-            if previous.pos_ in {"PROPN", "NOUN", "PRON"}:
+            if end >= len(doc):
+                break
+            next_token = doc[end]
+            if next_token.pos_ in {"PROPN", "NOUN", "PRON"}:
                 end += 1
             else:
                 break