From ef65db327557a7e8745038e05965de160288b113 Mon Sep 17 00:00:00 2001 From: svlandeg Date: Wed, 26 Jul 2023 16:24:34 +0200 Subject: [PATCH] use registry name if given and exit gracefully if a registry was not found --- spacy/cli/find_function.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/spacy/cli/find_function.py b/spacy/cli/find_function.py index 0053f190f..a594ad7b6 100644 --- a/spacy/cli/find_function.py +++ b/spacy/cli/find_function.py @@ -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)