edsnlp.pipelines.ner.scores.tnm.factory

DEFAULT_CONFIG = dict(pattern=None, attr='TEXT') module-attribute

create_component(nlp, name, pattern, attr)

Source code in edsnlp/pipelines/ner/scores/tnm/factory.py

@Language.factory(
    "eds.TNM",
    default_config=DEFAULT_CONFIG,
    assigns=["doc.ents", "doc.spans"],
)
def create_component(
    nlp: Language,
    name: str,
    pattern: Optional[Union[List[str], str]],
    attr: str,
):
    return TNM(
        nlp,
        pattern=pattern,
        attr=attr,
    )