Skip to content

Commit 40f15a7

Browse files
formatting (black)
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 53bc6ac commit 40f15a7

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

examples/experimental/sea_connector_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def test_sea_query_execution_with_compression():
7373
logger.info(
7474
f"Successfully opened SEA session with ID: {connection.get_session_id_hex()}"
7575
)
76-
76+
7777
# Execute a simple query with compression disabled
7878
cursor = connection.cursor(arraysize=0, buffer_size_bytes=0)
7979
logger.info("Executing query without compression: SELECT 1 as test_value")

src/databricks/sql/result_set.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -421,13 +421,15 @@ def __init__(
421421
# Legacy style with direct sea_response
422422
self._response = sea_response
423423
# Extract values from sea_response
424-
command_id = CommandId.from_sea_statement_id(sea_response.get("statement_id", ""))
424+
command_id = CommandId.from_sea_statement_id(
425+
sea_response.get("statement_id", "")
426+
)
425427
self.statement_id = sea_response.get("statement_id", "")
426-
428+
427429
# Extract status
428430
status_data = sea_response.get("status", {})
429431
status = CommandState.from_sea_state(status_data.get("state", "PENDING"))
430-
432+
431433
# Set defaults for other fields
432434
has_been_closed_server_side = False
433435
has_more_rows = False

0 commit comments

Comments
 (0)