explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-04-21 17:41:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

use registry name if given and exit gracefully if a registry was not found

Browse Source
This commit is contained in:
svlandeg 2023-07-26 16:24:34 +02:00
parent 7916d253c1
commit ef65db3275
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
spacy/cli/find_function.py
View File

@ -19,11 +19,18 @@ def find_function_cli(
func_name (str): Name of the registered function.
registry_name (str): Name of the catalogue registry.
"""
registry_names = registry.get_registry_names()
for name in registry_names:
if registry.has(name, func_name):
registry_name = name
break
if not registry_name:
registry_names = registry.get_registry_names()
for name in registry_names:
if registry.has(name, func_name):
registry_name = name
break
if not registry_name:
msg.fail(
f"Couldn't find registered function: {func_name}",
exits=1,
)
find_function(func_name, registry_name)