From 3ef72cf2a05eb2322089e37211c2046fad3b421f Mon Sep 17 00:00:00 2001 From: Ruifeng Zheng Date: Thu, 17 Jul 2025 20:24:43 +0800 Subject: [PATCH 1/3] test test nit --- .../connect/test_parity_classification.py | 38 +++++++++++++++++-- 1 file changed, 35 insertions(+), 3 deletions(-) diff --git a/python/pyspark/ml/tests/connect/test_parity_classification.py b/python/pyspark/ml/tests/connect/test_parity_classification.py index 7805546dba707..c43e784fd9a1c 100644 --- a/python/pyspark/ml/tests/connect/test_parity_classification.py +++ b/python/pyspark/ml/tests/connect/test_parity_classification.py @@ -17,14 +17,46 @@ import unittest +from pyspark.testing.utils import eventually, timeout from pyspark.ml.tests.test_classification import ClassificationTestsMixin from pyspark.testing.connectutils import ReusedConnectTestCase -# TODO(SPARK-52764): Re-enable this test after fixing the flakiness. -@unittest.skip("Disabled due to flakiness, should be enabled after fixing the issue") class ClassificationParityTests(ClassificationTestsMixin, ReusedConnectTestCase): - pass + @eventually(timeout=60, catch_timeout=True) + @timeout(timeout=10) + def test_binary_logistic_regression_summary(self): + super().test_binary_logistic_regression_summary() + + @eventually(timeout=60, catch_timeout=True) + @timeout(timeout=10) + def test_multiclass_logistic_regression_summary(self): + super().test_multiclass_logistic_regression_summary() + + @eventually(timeout=60, catch_timeout=True) + @timeout(timeout=10) + def test_linear_svc(self): + super().test_linear_svc() + + @eventually(timeout=60, catch_timeout=True) + @timeout(timeout=10) + def test_factorization_machine(self): + super().test_factorization_machine() + + @eventually(timeout=60, catch_timeout=True) + @timeout(timeout=10) + def test_binary_random_forest_classifier(self): + super().test_binary_random_forest_classifier() + + @eventually(timeout=60, catch_timeout=True) + @timeout(timeout=10) + def test_multiclass_random_forest_classifier(self): + super().test_multiclass_random_forest_classifier() + + @eventually(timeout=60, catch_timeout=True) + @timeout(timeout=10) + def test_mlp(self): + super().test_mlp() if __name__ == "__main__": From 3c22827e15da1cbbc3b4f033408d32c59fa32e4f Mon Sep 17 00:00:00 2001 From: Ruifeng Zheng Date: Fri, 18 Jul 2025 09:38:26 +0800 Subject: [PATCH 2/3] empty From 52773ee53608313b698fdc05c98a149e14bd64cd Mon Sep 17 00:00:00 2001 From: Ruifeng Zheng Date: Fri, 18 Jul 2025 12:07:10 +0800 Subject: [PATCH 3/3] re-trigger