From cb6c68a57371e9914271b57585b9e92e8cfb9fcf Mon Sep 17 00:00:00 2001 From: Ines Montani Date: Mon, 16 Sep 2019 15:16:29 +0200 Subject: [PATCH] Pass vectors name correctly in prune_vectors --- spacy/vocab.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/vocab.pyx b/spacy/vocab.pyx index 021da02fc..869c92921 100644 --- a/spacy/vocab.pyx +++ b/spacy/vocab.pyx @@ -318,7 +318,7 @@ cdef class Vocab: keys = xp.asarray([key for (prob, i, key) in priority], dtype="uint64") keep = xp.ascontiguousarray(self.vectors.data[indices[:nr_row]]) toss = xp.ascontiguousarray(self.vectors.data[indices[nr_row:]]) - self.vectors = Vectors(data=keep, keys=keys) + self.vectors = Vectors(data=keep, keys=keys, name=self.vectors.name) syn_keys, syn_rows, scores = self.vectors.most_similar(toss, batch_size=batch_size) remap = {} for i, key in enumerate(keys[nr_row:]):