increase tolerance in flaky test

piskvorky · Apr 29, 2023 · 5cc2cad · 5cc2cad
1 parent ccffa84
commit 5cc2cad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gensim/test/test_word2vec.py b/gensim/test/test_word2vec.py
@@ -847,9 +847,9 @@ def test_parallel(self):
             expected_neighbor = 'palestinian'
             sims = model.wv.most_similar(origin_word, topn=len(model.wv))
             # the exact vectors and therefore similarities may differ, due to different thread collisions/randomization
-            # so let's test only for top10
+            # so let's test only for topN
             neighbor_rank = [word for word, sim in sims].index(expected_neighbor)
-            self.assertLess(neighbor_rank, 5)
+            self.assertLess(neighbor_rank, 6)
 
     def test_r_n_g(self):
         """Test word2vec results identical with identical RNG seed."""