explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-06 05:10:21 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix type properly

Browse Source
This commit is contained in:
Kevin Humphreys 2022-08-26 01:30:44 +02:00
parent 63f5e1331d
commit 3dba984db9
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
spacy/pipeline/entityruler.py
View File

@ -26,9 +26,9 @@ PatternType = Dict[str, Union[str, List[Dict[str, Any]]]]
"phrase_matcher_attr": None, "phrase_matcher_attr": None,
"validate": False, "validate": False,
"overwrite_ents": False, "overwrite_ents": False,
"fuzzy": 0.0,
"ent_id_sep": DEFAULT_ENT_ID_SEP, "ent_id_sep": DEFAULT_ENT_ID_SEP,
"scorer": {"@scorers": "spacy.entity_ruler_scorer.v1"}, "scorer": {"@scorers": "spacy.entity_ruler_scorer.v1"},
"fuzzy": None,
}, },
default_score_weights={ default_score_weights={
"ents_f": 1.0, "ents_f": 1.0,
@ -43,9 +43,9 @@ def make_entity_ruler(
phrase_matcher_attr: Optional[Union[int, str]], phrase_matcher_attr: Optional[Union[int, str]],
validate: bool, validate: bool,
overwrite_ents: bool, overwrite_ents: bool,
fuzzy: float,
ent_id_sep: str, ent_id_sep: str,
scorer: Optional[Callable], scorer: Optional[Callable],
fuzzy: Optional[float],
): ):
return EntityRuler( return EntityRuler(
nlp, nlp,
@ -53,9 +53,9 @@ def make_entity_ruler(
phrase_matcher_attr=phrase_matcher_attr, phrase_matcher_attr=phrase_matcher_attr,
validate=validate, validate=validate,
overwrite_ents=overwrite_ents, overwrite_ents=overwrite_ents,
fuzzy=fuzzy,
ent_id_sep=ent_id_sep, ent_id_sep=ent_id_sep,
scorer=scorer, scorer=scorer,
fuzzy=fuzzy,
) )
@ -87,10 +87,10 @@ class EntityRuler(Pipe):
phrase_matcher_attr: Optional[Union[int, str]] = None, phrase_matcher_attr: Optional[Union[int, str]] = None,
validate: bool = False, validate: bool = False,
overwrite_ents: bool = False, overwrite_ents: bool = False,
fuzzy: float = 0,
ent_id_sep: str = DEFAULT_ENT_ID_SEP, ent_id_sep: str = DEFAULT_ENT_ID_SEP,
patterns: Optional[List[PatternType]] = None, patterns: Optional[List[PatternType]] = None,
scorer: Optional[Callable] = entity_ruler_score, scorer: Optional[Callable] = entity_ruler_score,
fuzzy: float = None,
) -> None: ) -> None:
"""Initialize the entity ruler. If patterns are supplied here, they """Initialize the entity ruler. If patterns are supplied here, they
need to be a list of dictionaries with a `"label"` and `"pattern"` need to be a list of dictionaries with a `"label"` and `"pattern"`
@ -122,7 +122,8 @@ class EntityRuler(Pipe):
self.token_patterns = defaultdict(list) # type: ignore self.token_patterns = defaultdict(list) # type: ignore
self.phrase_patterns = defaultdict(list) # type: ignore self.phrase_patterns = defaultdict(list) # type: ignore
self._validate = validate self._validate = validate
self.matcher = Matcher(nlp.vocab, validate=validate, fuzzy=fuzzy) self.fuzzy = fuzzy
self.matcher = Matcher(nlp.vocab, validate=validate, fuzzy=self.fuzzy)
self.phrase_matcher_attr = phrase_matcher_attr self.phrase_matcher_attr = phrase_matcher_attr
self.phrase_matcher = PhraseMatcher( self.phrase_matcher = PhraseMatcher(
nlp.vocab, attr=self.phrase_matcher_attr, validate=validate nlp.vocab, attr=self.phrase_matcher_attr, validate=validate
@ -132,7 +133,6 @@ class EntityRuler(Pipe):
if patterns is not None: if patterns is not None:
self.add_patterns(patterns) self.add_patterns(patterns)
self.scorer = scorer self.scorer = scorer
self.fuzzy = fuzzy
def __len__(self) -> int: def __len__(self) -> int:
"""The number of all patterns added to the entity ruler.""" """The number of all patterns added to the entity ruler."""