From ac8d9b39953dd7cff731e8966d720efb10d39068 Mon Sep 17 00:00:00 2001 From: Ruifeng Zheng Date: Thu, 23 Jan 2025 13:26:24 +0800 Subject: [PATCH] todo --- python/pyspark/ml/tests/test_feature.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/pyspark/ml/tests/test_feature.py b/python/pyspark/ml/tests/test_feature.py index 8aab46ad9c9e1..9766ab1b02438 100644 --- 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")