Make topn=None disable indexer in most_similar

piskvorky · Apr 23, 2019 · ad788ea · ad788ea
1 parent 6d89673
commit ad788ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gensim/models/keyedvectors.py b/gensim/models/keyedvectors.py
@@ -550,7 +550,7 @@ def most_similar(self, positive=None, negative=None, topn=10, restrict_vocab=Non
             raise ValueError("cannot compute similarity with no input")
         mean = matutils.unitvec(array(mean).mean(axis=0)).astype(REAL)
 
-        if indexer is not None:
+        if indexer is not None and isinstance(topn, int):
             return indexer.most_similar(mean, topn)
 
         limited = self.vectors_norm if restrict_vocab is None else self.vectors_norm[:restrict_vocab]