Fix test for Python 3

youndoldman · Oct 31, 2017 · 92dc127 · 92dc127
1 parent c5799ec
commit 92dc127
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spacy/tests/vocab/test_add_vectors.py b/spacy/tests/vocab/test_add_vectors.py
@@ -35,6 +35,6 @@ def test_vocab_prune_vectors():
 
     remap = vocab.prune_vectors(2)
     assert list(remap.keys()) == [u'kitten']
-    neighbour, similarity = remap.values()[0]
+    neighbour, similarity = list(remap.values())[0]
     assert neighbour == u'cat', remap
     assert_allclose(similarity, cosine(data[0], data[2]), atol=1e-6)