explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-14 18:22:27 +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(): for spec_type, spec in llm_spec.items():
valid_values = set()
user_value = spec["arg"].lower().replace(".", "-")
for reg_handle in spec["reg_handles"]: 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 spec["matched_reg_handle"] = reg_handle
break break
if not spec["matched_reg_handle"]: if not spec["matched_reg_handle"]:
arg = spec["arg"] arg = spec["arg"]
valid_args = sorted(spec["reg_handles"].keys())
msg.fail( msg.fail(
f"Couldn't find a matching registration handle for {spec_type} '{arg}'. " f"Couldn't find a matching registration handle for {spec_type} '{arg}'. "
f"Available functions: {valid_args}", f"Available functions: {valid_values}",
exits=1, exits=1,
) )