Class DocumentRater

document_rater.py:16–59

class DocumentRater:
    _name = None
    _require_doc_text = None

    def __init__(
        self, rater_name: str | None = None, normalizer: ScoreNormalizer | None = None, **kwargs
    ) -> None:
        self._obj_name = rater_name if rater_name is not None else self._name
        self.normalizer = normalizer
        logger.info(
            f"Initializing rater type: {self._name}, name: {self._obj_name}, normalizer: {self.normalizer}, args: {kwargs}"
        )

    def _annotate_doc(self, doc: Document, score: float | int) -> Document:
        if self.normalizer is None:
            return Document(
                docid=doc.docid,
                text=doc.text,
                annotations=DocumentAnnotation({self.get_name(): score, **doc.annotations}),
            )
        logger.debug(f"Normalizing score: {self.get_name()}")
        return Document(
            docid=doc.docid,
            text=doc.text,
            annotations=DocumentAnnotation(
                {
                    self.get_name(): score,
                    f"{self.get_name()}_{self.normalizer.get_name()}_normalized": self.normalizer(
                        score
                    ),
                    **doc.annotations,
                }
            ),
        )

    @abstractmethod
    def __call__(self, docs: list[Document]) -> list[Document]: ...

    def get_name(self) -> str | None:
        return self._obj_name

    @classmethod
    def require_doc_text(cls) -> bool | None:
        return cls._require_doc_text


class RandomRater(DocumentRater):