explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-11 08:42:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Test nested registered functions in overrides

Browse Source
This commit is contained in:
Adriane Boyd 2023-06-02 10:58:53 +02:00
parent b24573b349
commit 2e3c5be192
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
spacy/tests/serialize/test_serialize_config.py
View File

@ -10,6 +10,7 @@ from spacy.language import Language
from spacy.ml.models import MaxoutWindowEncoder, MultiHashEmbed from spacy.ml.models import MaxoutWindowEncoder, MultiHashEmbed
from spacy.ml.models import build_tb_parser_model, build_Tok2Vec_model from spacy.ml.models import build_tb_parser_model, build_Tok2Vec_model
from spacy.schemas import ConfigSchema, ConfigSchemaPretrain from spacy.schemas import ConfigSchema, ConfigSchemaPretrain
from spacy.training import Example
from spacy.util import load_config, load_config_from_str from spacy.util import load_config, load_config_from_str
from spacy.util import load_model_from_config, registry from spacy.util import load_model_from_config, registry
@ -434,6 +435,11 @@ def test_config_overrides_registered_functions():
nlp_re1.config["components"]["attribute_ruler"]["scorer"]["@scorers"] nlp_re1.config["components"]["attribute_ruler"]["scorer"]["@scorers"]
== "spacy.tagger_scorer.v1" == "spacy.tagger_scorer.v1"
) )
@registry.misc("test_some_other_key")
def misc_some_other_key():
return "some_other_key"
nlp_re2 = spacy.load( nlp_re2 = spacy.load(
d, d,
config={ config={
@ -441,7 +447,7 @@ def test_config_overrides_registered_functions():
"attribute_ruler": { "attribute_ruler": {
"scorer": { "scorer": {
"@scorers": "spacy.overlapping_labeled_spans_scorer.v1", "@scorers": "spacy.overlapping_labeled_spans_scorer.v1",
"spans_key": "some_other_key", "spans_key": {"@misc": "test_spans_key_value"},
} }
} }
} }
@ -449,8 +455,14 @@ def test_config_overrides_registered_functions():
) )
assert ( assert (
nlp_re2.config["components"]["attribute_ruler"]["scorer"]["spans_key"] nlp_re2.config["components"]["attribute_ruler"]["scorer"]["spans_key"]
== "some_other_key" == {"@misc": "test_some_other_key"}
) )
# run dummy evaluation (will return None scores) in order to test that
# the spans_key value in the nested override is working as intended in
# the config
example = Example.from_dict(nlp_re2.make_doc("a b c"), {})
scores = nlp_re2.evaluate([example])
assert "spans_some_other_key_f" in scores
def test_config_interpolation(): def test_config_interpolation():