diff --git a/python/pyspark/sql/connect/group.py b/python/pyspark/sql/connect/group.py index 0301cf0bd1f2b..b73834b145f1d 100644 --- a/python/pyspark/sql/connect/group.py +++ b/python/pyspark/sql/connect/group.py @@ -41,7 +41,7 @@ from pyspark.sql.column import Column from pyspark.sql.connect.functions import builtin as F from pyspark.errors import PySparkNotImplementedError, PySparkTypeError -from pyspark.sql.streaming.stateful_processor import StatefulProcessor, StatefulProcessorHandle +from pyspark.sql.streaming.stateful_processor import StatefulProcessor if TYPE_CHECKING: from pyspark.sql.connect._typing import ( diff --git a/python/pyspark/sql/connect/plan.py b/python/pyspark/sql/connect/plan.py index 1fec45088e477..59ca32f63f2ed 100644 --- a/python/pyspark/sql/connect/plan.py +++ b/python/pyspark/sql/connect/plan.py @@ -59,7 +59,6 @@ PySparkValueError, PySparkPicklingError, ) -from pyspark.sql.streaming.stateful_processor import StatefulProcessor if TYPE_CHECKING: from pyspark.sql.connect.client import SparkConnectClient diff --git a/python/pyspark/sql/pandas/group_ops.py b/python/pyspark/sql/pandas/group_ops.py index 5a03a7b98ff68..1fc63ce7b4171 100644 --- a/python/pyspark/sql/pandas/group_ops.py +++ b/python/pyspark/sql/pandas/group_ops.py @@ -15,7 +15,7 @@ # limitations under the License. # import sys -from typing import Any, Iterator, List, Optional, Union, TYPE_CHECKING, cast +from typing import List, Optional, Union, TYPE_CHECKING, cast import warnings from pyspark.errors import PySparkTypeError @@ -23,7 +23,7 @@ from pyspark.sql.column import Column from pyspark.sql.dataframe import DataFrame from pyspark.sql.streaming.state import GroupStateTimeout -from pyspark.sql.streaming.stateful_processor import StatefulProcessor, StatefulProcessorHandle +from pyspark.sql.streaming.stateful_processor import StatefulProcessor from pyspark.sql.types import StructType if TYPE_CHECKING: