explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-10-31 07:57:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix types

Browse Source
This commit is contained in:
Paul O'Leary McCann 2022-12-26 14:55:40 +09:00
parent ab2773e6b3
commit f3a928cb4b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/cli/configure.py
View File

@ -2,6 +2,7 @@ from pathlib import Path
import re import re
from wasabi import msg from wasabi import msg
import typer import typer
from thinc.api import Config
import spacy import spacy
from spacy.language import Language from spacy.language import Language
@ -162,7 +163,7 @@ def configure_resume_cli(
@configure_cli.command("transformer") @configure_cli.command("transformer")
def use_transformer( def use_transformer(
base_model: str, output_path: Path, transformer_name: str = "roberta-base" base_model: str, output_path: Path, transformer_name: str = "roberta-base"
): ) -> Config:
"""Replace pipeline tok2vec with transformer.""" """Replace pipeline tok2vec with transformer."""
# 1. identify tok2vec # 1. identify tok2vec
@ -206,7 +207,7 @@ def use_transformer(
@configure_cli.command("tok2vec") @configure_cli.command("tok2vec")
def use_tok2vec(base_model: str, output_path: Path) -> Language: def use_tok2vec(base_model: str, output_path: Path) -> Config:
"""Replace pipeline tok2vec with CNN tok2vec.""" """Replace pipeline tok2vec with CNN tok2vec."""
nlp = spacy.load(base_model) nlp = spacy.load(base_model)
_check_single_tok2vec(base_model, nlp.config) _check_single_tok2vec(base_model, nlp.config)