todo

apache · Jan 23, 2025 · ac8d9b3 · ac8d9b3
1 parent c426783
commit ac8d9b3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/pyspark/ml/tests/test_feature.py b/python/pyspark/ml/tests/test_feature.py
@@ -379,6 +379,7 @@ def test_word2vec(self):
         self.assertEqual(synonyms.columns, ["word", "similarity"])
         self.assertEqual(synonyms.count(), 2)
 
+        # TODO(SPARK-50958): Support Word2VecModel.findSynonymsArray
         # synonyms = model.findSynonymsArray("a", 2)
         # self.assertEqual(len(synonyms), 2)
         # self.assertEqual(synonyms[0][0], "b")