Function test_textcat_reduce_invalid_args

spacy/tests/test_models.py:290–301

()

def test_textcat_reduce_invalid_args():
    textcat_reduce = registry.architectures.get("spacy.TextCatReduce.v1")
    tok2vec = make_test_tok2vec()
    with pytest.raises(ValueError, match=r"must be used with at least one reduction"):
        textcat_reduce(
            tok2vec=tok2vec,
            exclusive_classes=False,
            use_reduce_first=False,
            use_reduce_last=False,
            use_reduce_max=False,
            use_reduce_mean=False,
        )