From 81276f21a52fa035935c1f96f59e7d9a77579bfb Mon Sep 17 00:00:00 2001 From: Paul O'Leary McCann Date: Mon, 13 Feb 2023 12:08:37 +0900 Subject: [PATCH] Change back to short names --- spacy/cli/__init__.py | 3 +-- spacy/cli/configure.py | 4 ++-- spacy/tests/test_cli.py | 3 +-- 3 files changed, 4 insertions(+), 6 deletions(-) diff --git a/spacy/cli/__init__.py b/spacy/cli/__init__.py index 003859777..47d05b5b6 100644 --- a/spacy/cli/__init__.py +++ b/spacy/cli/__init__.py @@ -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 diff --git a/spacy/cli/configure.py b/spacy/cli/configure.py index bfbcafb31..cd25ac351 100644 --- a/spacy/cli/configure.py +++ b/spacy/cli/configure.py @@ -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 diff --git a/spacy/tests/test_cli.py b/spacy/tests/test_cli.py index 7ede6339f..10701263f 100644 --- a/spacy/tests/test_cli.py +++ b/spacy/tests/test_cli.py @@ -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