fix: treat span as doc in Qualifier process method

Author: Thomzoy

edsnlp/pipes/qualifiers/base.py

@@ -172,9 +172,13 @@ def get_matches(self, doc: Doc) -> List[Span]:
 
         return list(matches)
 
-    def process(self, doc: Doc) -> BaseQualifierResults:
+    def ensure_doc(self, doc: Union[Doc, Span]) -> Doc:
+        return doc if not hasattr(doc, "as_doc") else doc.as_doc()
+
+    def process(self, doc_like: Union[Doc, Span]) -> BaseQualifierResults:
+        doc_like = self.ensure_doc(doc_like)  # pragma: no cover
         raise NotImplementedError
 
     def __call__(self, doc: Doc) -> Doc:
-        results = self.process(doc)
+        results = self.process(doc)  # pragma: no cover
         raise NotImplementedError(f"{type(results)} should be used to tag the document")

edsnlp/pipes/qualifiers/family/family.py

@@ -187,7 +187,8 @@ def set_extensions(self) -> None:
         if not Doc.has_extension("family"):
             Doc.set_extension("family", default=[])
 
-    def process(self, doc: Doc) -> FamilyResults:
+    def process(self, doc_like: Union[Doc, Span]) -> FamilyResults:
+        doc = self.ensure_doc(doc_like)
         matches = self.get_matches(doc)
 
         terminations = [m for m in matches if m.label_ == "termination"]

edsnlp/pipes/qualifiers/history/history.py

@@ -326,7 +326,8 @@ def set_extensions(self) -> None:
                 getter=deprecated_getter_factory("antecedent_cues", "history_cues"),
             )
 
-    def process(self, doc: Doc) -> HistoryResults:
+    def process(self, doc_like: Union[Doc, Span]) -> HistoryResults:
+        doc = self.ensure_doc(doc_like)
         note_datetime = None
         if doc._.note_datetime is not None:
             try:

edsnlp/pipes/qualifiers/hypothesis/hypothesis.py

@@ -262,7 +262,8 @@ def load_verbs(
             list_hypo_verbs_following,
         )
 
-    def process(self, doc: Doc) -> HypothesisResults:
+    def process(self, doc_like: Union[Doc, Span]) -> HypothesisResults:
+        doc = self.ensure_doc(doc_like)
         matches = self.get_matches(doc)
 
         terminations = [m for m in matches if m.label_ == "termination"]

edsnlp/pipes/qualifiers/negation/negation.py

@@ -295,7 +295,8 @@ def __call__(self, doc: Doc) -> Doc:
                     token._.negation = True
         return doc
 
-    def process(self, doc: Doc) -> NegationResults:
+    def process(self, doc_like: Union[Doc, Span]) -> NegationResults:
+        doc = self.ensure_doc(doc_like)
         matches = self.get_matches(doc)
 
         terminations = [m for m in matches if m.label_ == "termination"]

edsnlp/pipes/qualifiers/reported_speech/reported_speech.py

@@ -226,7 +226,8 @@ def load_verbs(self, verbs: List[str]) -> List[str]:
 
         return list_rep_verbs
 
-    def process(self, doc: Doc) -> ReportedSpeechResults:
+    def process(self, doc_like: Union[Doc, Span]) -> ReportedSpeechResults:
+        doc = self.ensure_doc(doc_like)
         matches = self.get_matches(doc)
         matches += list(self.regex_matcher(doc, as_spans=True))