From 36b47e3fa6ce94b567ba846115c7a058fddf0099 Mon Sep 17 00:00:00 2001
From: Matthew Honnibal <honnibal+gh@gmail.com>
Date: Thu, 7 Dec 2017 09:53:30 +0100
Subject: [PATCH] Fix (and test) vector pickling

---
 spacy/tests/test_pickles.py | 4 ++++
 spacy/vectors.pyx           | 1 +
 2 files changed, 5 insertions(+)

diff --git a/spacy/tests/test_pickles.py b/spacy/tests/test_pickles.py
index b22a2864b3d..f1118542af7 100644
--- a/spacy/tests/test_pickles.py
+++ b/spacy/tests/test_pickles.py
@@ -3,6 +3,7 @@
 
 import pytest
 import dill as pickle
+import numpy
 
 from ..vocab import Vocab
 from ..attrs import NORM
@@ -22,6 +23,7 @@ def test_pickle_string_store(stringstore, text1, text2):
 @pytest.mark.parametrize('text1,text2', [('dog', 'cat')])
 def test_pickle_vocab(text1, text2):
     vocab = Vocab(lex_attr_getters={int(NORM): lambda string: string[:-1]})
+    vocab.set_vector('dog', numpy.ones((5,), dtype='f'))
     lex1 = vocab[text1]
     lex2 = vocab[text2]
     assert lex1.norm_ == text1[:-1]
@@ -33,3 +35,5 @@ def test_pickle_vocab(text1, text2):
     assert unpickled[text1].norm == lex1.norm
     assert unpickled[text2].norm == lex2.norm
     assert unpickled[text1].norm != unpickled[text2].norm
+    assert unpickled.vectors is not None
+    assert list(vocab['dog'].vector) == [1.,1.,1.,1.,1.]
diff --git a/spacy/vectors.pyx b/spacy/vectors.pyx
index 1e4e97c3838..46997cccf93 100644
--- a/spacy/vectors.pyx
+++ b/spacy/vectors.pyx
@@ -19,6 +19,7 @@ def unpickle_vectors(keys_and_rows, data):
     vectors = Vectors(data=data)
     for key, row in keys_and_rows:
         vectors.add(key, row=row)
+    return vectors
 
 
 cdef class Vectors: