CU-8694ezhrn: Propgate keeping raw option properly

CogStack · Apr 30, 2024 · dd2c559 · dd2c559
1 parent fd29bfe
commit dd2c559
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/medcat/compare_models/compare.py b/medcat/compare_models/compare.py
@@ -46,12 +46,15 @@ def _get_pt2ch(cat: CAT) -> Optional[Dict]:
 
 
 def get_per_annotation_diffs(cat1: CAT, cat2: CAT, documents: Iterator[Tuple[str, str]],
-                             show_progress: bool = True) -> PerAnnotationDifferences:
+                             show_progress: bool = True,
+                             keep_raw: bool = True,
+                             ) -> PerAnnotationDifferences:
     pt2ch1: Optional[Dict] = _get_pt2ch(cat1)
     pt2ch2: Optional[Dict] = _get_pt2ch(cat2)
     pad = PerAnnotationDifferences(pt2ch1=pt2ch1, pt2ch2=pt2ch2,
                                    model1_cuis=set(cat1.cdb.cui2names),
-                                   model2_cuis=set(cat2.cdb.cui2names))
+                                   model2_cuis=set(cat2.cdb.cui2names),
+                                   keep_raw=keep_raw)
     for doc_id, doc in tqdm.tqdm(documents, disable=not show_progress):
         pad.look_at_doc(cat1.get_entities(doc), cat2.get_entities(doc), doc_id, doc)
     pad.finalise()