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

better representation of valid values

Browse Source
This commit is contained in:
svlandeg 2023-07-25 18:13:14 +02:00
parent 8da042ce17
commit 0948dc19fb
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
spacy/cli/init_config.py
View File

@ -208,19 +208,20 @@ def init_config(
}
for spec_type, spec in llm_spec.items():
valid_values = set()
user_value = spec["arg"].lower().replace(".", "-")
for reg_handle in spec["reg_handles"]:
if reg_handle.split(".")[1].lower() == spec["arg"].lower().replace(
".", "-"
):
reg_name = reg_handle.split(".")[1]
valid_values.add(reg_name)
if reg_name.lower() == user_value:
spec["matched_reg_handle"] = reg_handle
break
if not spec["matched_reg_handle"]:
arg = spec["arg"]
valid_args = sorted(spec["reg_handles"].keys())
msg.fail(
f"Couldn't find a matching registration handle for {spec_type} '{arg}'. "
f"Available functions: {valid_args}",
f"Available functions: {valid_values}",
exits=1,
)