From a13c4e4f3c200facba42e219ac0f1dcb48e1b505 Mon Sep 17 00:00:00 2001 From: jingz-db Date: Thu, 30 Jan 2025 14:28:36 -0800 Subject: [PATCH] linter --- python/pyspark/sql/connect/group.py | 2 +- python/pyspark/sql/connect/plan.py | 1 - python/pyspark/sql/pandas/group_ops.py | 4 ++-- 3 files changed, 3 insertions(+), 4 deletions(-) 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: