explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-06-10 08:03:15 +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 = """ lazy_init_model_config = """
[model] [model]
@architectures = "LazyInitTok2Vec.v1" @architectures = "test.LazyInitTok2Vec.v1"
width = 96 width = 96
""" """
LAZY_INIT_TOK2VEC_MODEL = Config().from_str(lazy_init_model_config)["model"] 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 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]]: def build_lazy_init_tok2vec(*, width: int) -> Model[List[Doc], List[Floats2d]]:
"""tok2vec model of which the output size is only known after """tok2vec model of which the output size is only known after
initialization. This implementation does not output meaningful initialization. This implementation does not output meaningful