From b2a7c9478723d986d29fac3421aa7df71ef3fb13 Mon Sep 17 00:00:00 2001 From: svlandeg Date: Wed, 26 Jul 2023 17:17:48 +0200 Subject: [PATCH] more mypy fixes --- spacy/cli/find_function.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/spacy/cli/find_function.py b/spacy/cli/find_function.py index b00d50485..9f12d2259 100644 --- a/spacy/cli/find_function.py +++ b/spacy/cli/find_function.py @@ -44,22 +44,26 @@ def find_function(func_name: str, registry_name: str) -> Tuple[str, int]: registry_desc = None try: registry_desc = registry.find(registry_name, func_name) - except RegistryError as e: msg.fail( f"Couldn't find registered function: '{func_name}' in registry '{registry_name}'", ) msg.fail(f"{e}", exits=1) - assert registry_desc is not None + + registry_path = None + line_no = None if registry_desc["file"]: registry_path = registry_desc["file"] line_no = registry_desc["line_no"] - else: + + if not registry_path or not line_no: msg.fail( f"Couldn't find path to registered function: '{func_name}' in registry '{registry_name}'", exits=1, ) + assert registry_path is not None + assert line_no is not None msg.good(f"Found registered function at {registry_path}:{line_no}") - return registry_path, line_no + return str(registry_path), int(line_no)