fix typo

piskvorky · Jul 30, 2020 · cbc755e · cbc755e
1 parent d40f685
commit cbc755e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/gensim/models/keyedvectors.py b/gensim/models/keyedvectors.py
@@ -358,10 +358,10 @@ def get_vector(self, key, norm=False):
 
         Parameters
         ----------
-        key : str or int
-            Key for vector to return, or int slot
+        key : str
+            Key for vector to return.
         norm : bool, optional
-            If True - resulting vector will be L2-normalized (unit euclidean length).
+            If True, the resulting vector will be L2-normalized (unit euclidean length).
 
         Returns
         -------

diff --git a/gensim/test/test_word2vec.py b/gensim/test/test_word2vec.py
@@ -712,7 +712,7 @@ def testTrainingSgNegative(self):
         # self.assertTrue(sims[0][0] == 'trees', sims)  # most similar
 
         # test querying for "most similar" by vector
-        graph_vector = model.wv.get_vectors('graph', norm=True)
+        graph_vector = model.wv.get_vector('graph', norm=True)
         sims2 = model.wv.most_similar(positive=[graph_vector], topn=11)
         sims2 = [(w, sim) for w, sim in sims2 if w != 'graph']  # ignore 'graph' itself
         self.assertEqual(sims, sims2)