fix a test for oov similarity

piskvorky · Mar 3, 2020 · e7f555c · e7f555c
1 parent 1cc6c4a
commit e7f555c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gensim/test/test_fasttext.py b/gensim/test/test_fasttext.py
@@ -330,8 +330,8 @@ def test_oov_similarity(self):
         word = 'someoovword'
         most_similar = self.test_model.wv.most_similar(word)
         top_neighbor, top_similarity = most_similar[0]
-        v1 = self.test_model.vw[word]
-        v2 = self.test_model.vw[top_neighbor]
+        v1 = self.test_model.wv[word]
+        v2 = self.test_model.wv[top_neighbor]
         top_similarity_direct = self.test_model.wv.cosine_similarity(v1, v2.reshape(1, -1))[0]
         self.assertAlmostEqual(top_similarity, top_similarity_direct)