From 27a4925f8db14c1cf2de2715ea2f75d7ba71ff20 Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Fri, 2 Dec 2022 08:42:12 +0100 Subject: [PATCH] Add FUZZY6..9 --- spacy/matcher/matcher.pyx | 4 ++++ spacy/schemas.py | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/spacy/matcher/matcher.pyx b/spacy/matcher/matcher.pyx index 2e41d24a5..9316d3e7c 100644 --- a/spacy/matcher/matcher.pyx +++ b/spacy/matcher/matcher.pyx @@ -1035,6 +1035,10 @@ def _get_extra_predicates(spec, extra_predicates, vocab, fuzzy_compare): "FUZZY3": _FuzzyPredicate, "FUZZY4": _FuzzyPredicate, "FUZZY5": _FuzzyPredicate, + "FUZZY6": _FuzzyPredicate, + "FUZZY7": _FuzzyPredicate, + "FUZZY8": _FuzzyPredicate, + "FUZZY9": _FuzzyPredicate, } seen_predicates = {pred.key: pred.i for pred in extra_predicates} output = [] diff --git a/spacy/schemas.py b/spacy/schemas.py index 1c5c6871e..c7bc8d533 100644 --- a/spacy/schemas.py +++ b/spacy/schemas.py @@ -168,6 +168,10 @@ class TokenPatternString(BaseModel): FUZZY3: Optional[Union[StrictStr, "TokenPatternString"]] = Field(None, alias="fuzzy3") FUZZY4: Optional[Union[StrictStr, "TokenPatternString"]] = Field(None, alias="fuzzy4") FUZZY5: Optional[Union[StrictStr, "TokenPatternString"]] = Field(None, alias="fuzzy5") + FUZZY6: Optional[Union[StrictStr, "TokenPatternString"]] = Field(None, alias="fuzzy5") + FUZZY7: Optional[Union[StrictStr, "TokenPatternString"]] = Field(None, alias="fuzzy5") + FUZZY8: Optional[Union[StrictStr, "TokenPatternString"]] = Field(None, alias="fuzzy5") + FUZZY9: Optional[Union[StrictStr, "TokenPatternString"]] = Field(None, alias="fuzzy5") class Config: extra = "forbid"