From b2a2892c1a221ac37bc81c76a982e0419bcdf674 Mon Sep 17 00:00:00 2001 From: Dongjoon Hyun Date: Thu, 13 Nov 2025 08:10:48 -0800 Subject: [PATCH] [SPARK-54194][PYTHON][FOLLOWUP] Fix `connectutils.py` to import `pb2` conditionally --- python/pyspark/testing/connectutils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/pyspark/testing/connectutils.py b/python/pyspark/testing/connectutils.py index d895c1d8a26b1..302ae2a5751f7 100644 --- a/python/pyspark/testing/connectutils.py +++ b/python/pyspark/testing/connectutils.py @@ -23,7 +23,6 @@ import uuid import contextlib -import pyspark.sql.connect.proto as pb2 from pyspark import Row, SparkConf from pyspark.util import is_remote_only from pyspark.testing.utils import PySparkErrorTestUtils @@ -53,6 +52,7 @@ from pyspark.sql.connect.dataframe import DataFrame from pyspark.sql.connect.plan import Read, Range, SQL, LogicalPlan from pyspark.sql.connect.session import SparkSession + import pyspark.sql.connect.proto as pb2 class MockRemoteSession: