add result content to pregnancy task

ClarityNLP · Dec 9, 2019 · 7a33b2e · 7a33b2e
1 parent 4ddc657
commit 7a33b2e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nlp/custom_tasks/PregnancyTask.py b/nlp/custom_tasks/PregnancyTask.py
@@ -283,7 +283,8 @@ def run_custom_task(self, temp_file, mongo_client: MongoClient):
                         'sentence': sentence,
                         'highlights': highlights,
                         'start':[start],
-                        'end':[end]
+                        'end':[end],
+                        'result_content':sentence
                     }
                 }