explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 01:46:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Raise RegistryError

Browse Source
This commit is contained in:
Ines Montani 2021-01-16 12:57:13 +11:00
parent d1338966ae
commit d12be459f6
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/util.py
View File

@ -15,6 +15,7 @@ import numpy.random
import numpy import numpy
import srsly import srsly
import catalogue import catalogue
from catalogue import RegistryError
import sys import sys
import warnings import warnings
from packaging.specifiers import SpecifierSet, InvalidSpecifier from packaging.specifiers import SpecifierSet, InvalidSpecifier
@ -111,11 +112,11 @@ class registry(thinc.registry):
# We're overwriting this classmethod so we're able to provide more # We're overwriting this classmethod so we're able to provide more
# specific error messages and implement a fallback to spacy-legacy. # specific error messages and implement a fallback to spacy-legacy.
if not hasattr(cls, registry_name): if not hasattr(cls, registry_name):
raise ValueError(Errors.E894.format(name=registry_name)) raise RegistryError(Errors.E894.format(name=registry_name))
reg = getattr(cls, registry_name) reg = getattr(cls, registry_name)
try: try:
func = reg.get(func_name) func = reg.get(func_name)
except catalogue.RegistryError: except RegistryError:
if func_name.startswith("spacy."): if func_name.startswith("spacy."):
legacy_name = func_name.replace("spacy.", "spacy-legacy.") legacy_name = func_name.replace("spacy.", "spacy-legacy.")
try: try:
@ -123,7 +124,7 @@ class registry(thinc.registry):
except catalogue.RegistryError: except catalogue.RegistryError:
pass pass
available = ", ".join(sorted(reg.get_all().keys())) or "none" available = ", ".join(sorted(reg.get_all().keys())) or "none"
raise ValueError( raise RegistryError(
Errors.E893.format( Errors.E893.format(
name=func_name, reg_name=registry_name, available=available name=func_name, reg_name=registry_name, available=available
) )