resolvede empty coref issues #3

davidberenstein1957 · Mar 31, 2022 · 29faf81 · 29faf81
1 parent 4b3cb1c
commit 29faf81
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -129,4 +129,5 @@ dmypy.json
 .pyre/
 /models
 *.tar.gz
-/crosslingual_coreference/models
+/crosslingual_coreference/models
+test.py
diff --git a/crosslingual_coreference/CrossLingualPredictor.py b/crosslingual_coreference/CrossLingualPredictor.py
@@ -78,7 +78,8 @@ def predict(self, text: str, advanced_resolve: bool = True) -> dict:
 
         # Passing a document with no coreferences returns its original form
         if not clusters:
-            return text
+            prediction['resolved_text'] = text
+            return prediction
 
         doc = self.predictor._spacy(text)
 

diff --git a/crosslingual_coreference/examples/test_spacy.py b/crosslingual_coreference/examples/test_spacy.py
@@ -5,7 +5,6 @@
 
 nlp = spacy.load('nl_core_news_sm')
 
-nlp.remove_pipe('xx_coref')
 nlp.add_pipe('xx_coref', config={"model_name": "xlm_roberta"})
 
 for doc in nlp.pipe(texts):