explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-10-30 23:47:31 +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
from wasabi import msg
import typer
from thinc.api import Config
import spacy
from spacy.language import Language
@ -162,7 +163,7 @@ def configure_resume_cli(
@configure_cli.command("transformer")
def use_transformer(
base_model: str, output_path: Path, transformer_name: str = "roberta-base"
):
) -> Config:
"""Replace pipeline tok2vec with transformer."""
# 1. identify tok2vec
@ -206,7 +207,7 @@ def use_transformer(
@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."""
nlp = spacy.load(base_model)
_check_single_tok2vec(base_model, nlp.config)