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

Add test. prefix to LazyInitTok2Vec.v1

Browse Source
This commit is contained in:
Daniël de Kok 2024-02-02 12:10:20 +01:00
parent 0e19d45cce
commit 30a7171e58
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/tests/pipeline/test_textcat.py
View File

@ -44,7 +44,7 @@ TRAIN_DATA_MULTI_LABEL = [
lazy_init_model_config = """
[model]
@architectures = "LazyInitTok2Vec.v1"
@architectures = "test.LazyInitTok2Vec.v1"
width = 96
"""
LAZY_INIT_TOK2VEC_MODEL = Config().from_str(lazy_init_model_config)["model"]

2
spacy/tests/tok2vec.py
View File

@ -7,7 +7,7 @@ from spacy.tokens import Doc
from spacy.util import registry
@registry.architectures("LazyInitTok2Vec.v1")
@registry.architectures("test.LazyInitTok2Vec.v1")
def build_lazy_init_tok2vec(*, width: int) -> Model[List[Doc], List[Floats2d]]:
"""tok2vec model of which the output size is only known after
initialization. This implementation does not output meaningful