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

Change back to short names

Browse Source
This commit is contained in:
Paul O'Leary McCann 2023-02-13 12:08:37 +09:00
parent 4279c73b6d
commit 81276f21a5
3 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/cli/__init__.py
View File

@ -29,8 +29,7 @@ from .project.push import project_push # noqa: F401
from .project.pull import project_pull # noqa: F401
from .project.document import project_document # noqa: F401
from .find_threshold import find_threshold # noqa: F401
from .configure import configure_tok2vec_feature_source # noqa: F401
from .configure import configure_transformer_feature_source # noqa: F401
from .configure import use_transformer, use_tok2vec # noqa: F401
from .configure import configure_resume_cli # noqa: F401
from .merge import merge_pipelines # noqa: F401

4
spacy/cli/configure.py
View File

@ -126,7 +126,7 @@ def configure_resume_cli(
@configure_cli.command("transformer")
def configure_transformer_feature_source(
def use_transformer(
base_model: str, output_file: Path, transformer_name: str = "roberta-base"
) -> Config:
"""Replace pipeline tok2vec with transformer.
@ -180,7 +180,7 @@ def configure_transformer_feature_source(
@configure_cli.command("tok2vec")
def configure_tok2vec_feature_source(base_model: str, output_file: Path) -> Config:
def use_tok2vec(base_model: str, output_file: Path) -> Config:
"""Replace pipeline tok2vec with CNN tok2vec.
DOCS: https://spacy.io/api/cli#configure-tok2vec

3
spacy/tests/test_cli.py
View File

@ -21,8 +21,7 @@ from spacy.cli._util import parse_config_overrides, string_to_list
from spacy.cli._util import substitute_project_variables
from spacy.cli._util import validate_project_commands
from spacy.cli._util import upload_file, download_file
from spacy.cli.configure import configure_resume_cli
from spacy.cli.configure import configure_tok2vec_feature_source
from spacy.cli.configure import configure_resume_cli, use_tok2vec
from spacy.cli.debug_data import _compile_gold, _get_labels_from_model
from spacy.cli.debug_data import _get_labels_from_spancat
from spacy.cli.debug_data import _get_distribution, _get_kl_divergence