update usages

piskvorky · Dec 8, 2019 · a0ee1b0 · a0ee1b0
1 parent 12822f8
commit a0ee1b0
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/gensim/test/test_similarities.py b/gensim/test/test_similarities.py
@@ -547,7 +547,7 @@ def setUp(self):
 
  def testWord2Vec(self):
  model = word2vec.Word2Vec(texts, min_count=1)
- model.init_sims()
+ model.wv.init_sims()
  index = self.indexer(model, 10)
 
  self.assertVectorIsSimilarToItself(model.wv, index)
@@ -710,7 +710,7 @@ def setUp(self):
 
  def test_word2vec(self):
  model = word2vec.Word2Vec(texts, min_count=1)
- model.init_sims()
+ model.wv.init_sims()
  index = self.indexer(model)
 
  self.assertVectorIsSimilarToItself(model.wv, index)
@@ -729,11 +729,11 @@ def __iter__(self):
  yield line.lower().strip().split()
 
  model = FastText(LeeReader(datapath('lee.cor')))
- model.init_sims()
+ model.wv.init_sims()
  index = self.indexer(model)
 
  self.assertVectorIsSimilarToItself(model.wv, index)
- self.assertApproxNeighborsMatchExact(model, model.wv, index)
+ self.assertApproxNeighborsMatchExact(model.wv, model.wv, index)
  self.assertIndexSaved(index)
  self.assertLoadedIndexEqual(index, model)
 
@@ -801,7 +801,7 @@ def setUp(self):
  from gensim.similarities.nmslib import NmslibIndexer
 
  self.model = doc2vec.Doc2Vec(sentences, min_count=1)
- self.model.init_sims()
+ self.model.docvecs.init_sims()
  self.index = NmslibIndexer(self.model)
  self.vector = self.model.docvecs.vectors_docs_norm[0]