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

use msg.fail instead of raising error to provide nicer output on the console

Browse Source
This commit is contained in:
svlandeg 2023-07-25 18:05:14 +02:00
parent 9cf8ac87be
commit 8da042ce17
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
spacy/cli/init_config.py
View File

@ -187,12 +187,14 @@ def init_config(
raise ValueError(Errors.E1055) from ex raise ValueError(Errors.E1055) from ex
if llm_model is None: if llm_model is None:
raise ValueError( msg.fail(
"Option `llm.model` must be set if `llm` component is in pipeline." "Option `--llm.model` must be set if `llm` component is in pipeline.",
exits=1,
) )
if llm_task is None: if llm_task is None:
raise ValueError( msg.fail(
"Option `llm.task` must be set if `llm` component is in pipeline." "Option `--llm.task` must be set if `llm` component is in pipeline.",
exits=1,
) )
# Select registry handles for model(s) and task(s). Raise if no match found. # Select registry handles for model(s) and task(s). Raise if no match found.
@ -215,9 +217,11 @@ def init_config(
if not spec["matched_reg_handle"]: if not spec["matched_reg_handle"]:
arg = spec["arg"] arg = spec["arg"]
raise ValueError( valid_args = sorted(spec["reg_handles"].keys())
f"Couldn't find a matching registration handle for {spec_type} '{spec}'. Double-check" msg.fail(
f" whether '{arg}' is spelled correctly." f"Couldn't find a matching registration handle for {spec_type} '{arg}'. "
f"Available functions: {valid_args}",
exits=1,
) )
defaults = RECOMMENDATIONS["__default__"] defaults = RECOMMENDATIONS["__default__"]