explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-24 17:06:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Allow sourcing disabled components (#7215)

Browse Source 
Check `component_names` instead of `pipe_names` to allow sourcing
disabled components.
This commit is contained in:
Adriane Boyd 2021-02-26 13:50:56 +01:00 committed by GitHub
parent 10c930cc96
commit e43d43db32
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/language.py
View File

@ -684,12 +684,12 @@ class Language:
# TODO: handle errors and mismatches (vectors etc.)
if not isinstance(source, self.__class__):
raise ValueError(Errors.E945.format(name=source_name, source=type(source)))
if not source.has_pipe(source_name):
if not source_name in source.component_names:
raise KeyError(
Errors.E944.format(
name=source_name,
model=f"{source.meta['lang']}_{source.meta['name']}",
opts=", ".join(source.pipe_names),
opts=", ".join(source.component_names),
)
)
pipe = source.get_pipe(source_name)