Update slow readers test to use textcat_multilabel (#9300)

This commit is contained in:
Adriane Boyd 2021-09-27 20:04:02 +02:00
parent 694c318f4f
commit ebcc7d830f

View File

@ -82,15 +82,15 @@ def test_cat_readers(reader, additional_config):
[nlp] [nlp]
lang = "en" lang = "en"
pipeline = ["tok2vec", "textcat"] pipeline = ["tok2vec", "textcat_multilabel"]
[components] [components]
[components.tok2vec] [components.tok2vec]
factory = "tok2vec" factory = "tok2vec"
[components.textcat] [components.textcat_multilabel]
factory = "textcat" factory = "textcat_multilabel"
""" """
config = Config().from_str(nlp_config_string) config = Config().from_str(nlp_config_string)
fix_random_seed(config["training"]["seed"]) fix_random_seed(config["training"]["seed"])