From 1f58c5e314c2bae0d6834ef88bbf5f76040703ae Mon Sep 17 00:00:00 2001 From: Weichen Xu Date: Tue, 7 May 2024 11:28:42 +0800 Subject: [PATCH] init Signed-off-by: Weichen Xu --- python/pyspark/ml/tests/connect/test_connect_classification.py | 1 - python/pyspark/ml/tests/connect/test_connect_evaluation.py | 1 - 2 files changed, 2 deletions(-) diff --git a/python/pyspark/ml/tests/connect/test_connect_classification.py b/python/pyspark/ml/tests/connect/test_connect_classification.py index 2763d3f613ae8..f0d60a117e12f 100644 --- a/python/pyspark/ml/tests/connect/test_connect_classification.py +++ b/python/pyspark/ml/tests/connect/test_connect_classification.py @@ -21,7 +21,6 @@ from pyspark.sql import SparkSession from pyspark.ml.tests.connect.test_legacy_mode_classification import ClassificationTestsMixin -# TODO(SPARK-48083): Reenable this test case have_torch = "SPARK_SKIP_CONNECT_COMPAT_TESTS" not in os.environ try: import torch # noqa: F401 diff --git a/python/pyspark/ml/tests/connect/test_connect_evaluation.py b/python/pyspark/ml/tests/connect/test_connect_evaluation.py index 35af54605ca81..58076dfe0bbe6 100644 --- a/python/pyspark/ml/tests/connect/test_connect_evaluation.py +++ b/python/pyspark/ml/tests/connect/test_connect_evaluation.py @@ -20,7 +20,6 @@ from pyspark.sql import SparkSession from pyspark.ml.tests.connect.test_legacy_mode_evaluation import EvaluationTestsMixin -# TODO(SPARK-48084): Reenable this test case have_torcheval = "SPARK_SKIP_CONNECT_COMPAT_TESTS" not in os.environ try: import torcheval # noqa: F401