From 79f11d4f854dbf283e765d4f072d13b502399413 Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <honnibal+gh@gmail.com>
Date: Thu, 23 Nov 2017 17:19:50 +0100
Subject: [PATCH] Pickle vectors with vocab

---
 spacy/vocab.pyx | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/spacy/vocab.pyx b/spacy/vocab.pyx
index 59df8cc1b..438a08ca6 100644
--- a/spacy/vocab.pyx
+++ b/spacy/vocab.pyx
@@ -470,26 +470,27 @@ cdef class Vocab:
 
 def pickle_vocab(vocab):
     sstore = vocab.strings
+    vectors = vocab.vectors
     morph = vocab.morphology
     length = vocab.length
     data_dir = vocab.data_dir
     lex_attr_getters = dill.dumps(vocab.lex_attr_getters)
     lexemes_data = vocab.lexemes_to_bytes()
     return (unpickle_vocab,
-            (sstore, morph, data_dir, lex_attr_getters, lexemes_data, length))
+            (sstore, vectors, morph, data_dir, lex_attr_getters, lexemes_data, length))
 
 
-def unpickle_vocab(sstore, morphology, data_dir,
+def unpickle_vocab(sstore, vectors, morphology, data_dir,
                    lex_attr_getters, bytes lexemes_data, int length):
     cdef Vocab vocab = Vocab()
     vocab.length = length
+    vocab.vectors = vectors
     vocab.strings = sstore
     vocab.morphology = morphology
     vocab.data_dir = data_dir
     vocab.lex_attr_getters = dill.loads(lex_attr_getters)
     vocab.lexemes_from_bytes(lexemes_data)
     vocab.length = length
-    link_vectors_to_models(vocab)
     return vocab