From f3a928cb4b8967a0077c1762eced7932c5be1c3d Mon Sep 17 00:00:00 2001 From: Paul O'Leary McCann Date: Mon, 26 Dec 2022 14:55:40 +0900 Subject: [PATCH] Fix types --- spacy/cli/configure.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/spacy/cli/configure.py b/spacy/cli/configure.py index aa28da480..2d0f1ade2 100644 --- a/spacy/cli/configure.py +++ b/spacy/cli/configure.py @@ -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)