Skip to content

Commit a45eb2b

Browse files
[SPARK-50956][ML][TESTS][FOLLOW-UP] Skip flaky EvaluationTestsOnConnect
### What changes were proposed in this pull request? Skip flaky EvaluationTestsOnConnect ### Why are the changes needed? it is not stable ### Does this PR introduce _any_ user-facing change? No, test-only ### How was this patch tested? CI ### Was this patch authored or co-authored using generative AI tooling? No Closes #49666 from zhengruifeng/skip_connect_eval. Lead-authored-by: Ruifeng Zheng <[email protected]> Co-authored-by: Hyukjin Kwon <[email protected]> Signed-off-by: Ruifeng Zheng <[email protected]>
1 parent aa9a104 commit a45eb2b

File tree

1 file changed

+2
-14
lines changed

1 file changed

+2
-14
lines changed

python/pyspark/ml/tests/connect/test_connect_evaluation.py

Lines changed: 2 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -19,16 +19,12 @@
1919
import unittest
2020

2121
from pyspark.sql import SparkSession
22-
from pyspark.testing.connectutils import should_test_connect, connect_requirement_message
23-
from pyspark.testing.utils import have_torcheval, torcheval_requirement_message
22+
from pyspark.testing.connectutils import should_test_connect
2423

2524
if should_test_connect:
2625
from pyspark.ml.tests.connect.test_legacy_mode_evaluation import EvaluationTestsMixin
2726

28-
@unittest.skipIf(
29-
not should_test_connect or not have_torcheval,
30-
connect_requirement_message or torcheval_requirement_message,
31-
)
27+
@unittest.skip("SPARK-50956: Flaky with RetriesExceeded")
3228
class EvaluationTestsOnConnect(EvaluationTestsMixin, unittest.TestCase):
3329
def setUp(self) -> None:
3430
self.spark = SparkSession.builder.remote(
@@ -38,14 +34,6 @@ def setUp(self) -> None:
3834
def tearDown(self) -> None:
3935
self.spark.stop()
4036

41-
@unittest.skip("SPARK-50913: Flaky with RetriesExceeded")
42-
def test_binary_classifier_evaluator(self):
43-
self.test_binary_classifier_evaluator()
44-
45-
@unittest.skip("SPARK-50956: Flaky with RetriesExceeded")
46-
def test_multiclass_classifier_evaluator(self):
47-
self.test_multiclass_classifier_evaluator()
48-
4937

5038
if __name__ == "__main__":
5139
from pyspark.ml.tests.connect.test_connect_evaluation import * # noqa: F401,F403

0 commit comments

Comments
 (0)