Merge pull request #77 from joeyism/master

Updated to fix E966 in spacy
HHousen · Dec 12, 2022 · c7e7180 · c7e7180
2 parents 15bd11d + c7a8863
commit c7e7180
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/extractive.py b/src/extractive.py
@@ -1098,7 +1098,14 @@ def predict_sentences(
         else:
             nlp = English()
             sentencizer = nlp.create_pipe("sentencizer")
-            nlp.add_pipe(sentencizer)
+            try:
+                nlp.add_pipe(sentencizer)
+            except ValueError as e:
+                if e.args[0].startswith("[E966]"):
+                    nlp.add_pipe("sentencizer")
+                else:
+                    raise e
+
 
             src_txt = [
                 " ".join([token.text for token in nlp(sentence) if str(token) != "."])