This is an automated email from the ASF dual-hosted git repository. kabhwan pushed a commit to branch WIP-python-data-source-admission-control-trigger-availablenow-change-the-method-signature in repository https://gitbox.apache.org/repos/asf/spark.git
commit 213da610a759219ddbe28ba0b4a7bad08abde879 Author: Jungtaek Lim <[email protected]> AuthorDate: Wed Jan 21 17:45:16 2026 +0900 fix test --- .../pyspark/sql/tests/test_python_streaming_datasource.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/python/pyspark/sql/tests/test_python_streaming_datasource.py b/python/pyspark/sql/tests/test_python_streaming_datasource.py index 911f7fc81007..15f876bd5d93 100644 --- a/python/pyspark/sql/tests/test_python_streaming_datasource.py +++ b/python/pyspark/sql/tests/test_python_streaming_datasource.py @@ -26,9 +26,15 @@ from pyspark.sql.datasource import ( DataSourceStreamWriter, DataSourceStreamArrowWriter, SimpleDataSourceStreamReader, - SupportsTriggerAvailableNow, WriterCommitMessage, ) +from pyspark.sql.streaming.datasource import ( + ReadAllAvailable, + ReadLimit, + ReadMaxRows, + SupportsAdmissionControl, + SupportsTriggerAvailableNow, +) from pyspark.sql.streaming import StreamingQueryException from pyspark.sql.types import Row from pyspark.testing.sqlutils import ( @@ -141,13 +147,6 @@ class BasePythonStreamingDataSourceTestsMixin: return TestDataSource def _get_test_data_source_for_admission_control(self): - from pyspark.sql.streaming.datasource import ( - ReadAllAvailable, - ReadLimit, - ReadMaxRows, - SupportsAdmissionControl, - ) - class TestDataStreamReader(DataSourceStreamReader, SupportsAdmissionControl): def initialOffset(self): return {"partition-1": 0} --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
