fix

Author: zhengruifeng

python/pyspark/ml/util.py

@@ -40,7 +40,6 @@
 from contextlib import contextmanager
 
 from pyspark import since
-from pyspark.errors.exceptions.connect import SparkException
 from pyspark.ml.common import inherit_doc
 from pyspark.sql import SparkSession
 from pyspark.sql.utils import is_remote
@@ -319,6 +318,8 @@ def remote_call() -> Any:
                 return deserialize(properties)
 
         if is_remote() and "PYSPARK_NO_NAMESPACE_SHARE" not in os.environ:
+            from pyspark.errors.exceptions.connect import SparkException
+
             try:
                 return remote_call()
             except SparkException as e: