diff --git a/spacy/tests/training/test_pretraining.py b/spacy/tests/training/test_pretraining.py index 30b93e303..d1db92de5 100644 --- a/spacy/tests/training/test_pretraining.py +++ b/spacy/tests/training/test_pretraining.py @@ -360,7 +360,9 @@ def test_pretrain_default_vectors(): create_pretrain_vectors(1, 1, "cosine")(nlp.vocab, nlp.get_pipe("tok2vec").model) # floret vectors are supported - nlp.vocab.vectors = Vectors(data=get_current_ops().xp.zeros((10, 10)), mode="floret", hash_count=1) + nlp.vocab.vectors = Vectors( + data=get_current_ops().xp.zeros((10, 10)), mode="floret", hash_count=1 + ) create_pretrain_vectors(1, 1, "cosine")(nlp.vocab, nlp.get_pipe("tok2vec").model) # error for no vectors