Skip to content

Commit d78ffba

Browse files
committed
comment fix
Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>
1 parent a35d3e9 commit d78ffba

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

src/databricks/sql/backend/sea/queue.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -136,11 +136,12 @@ def __init__(
136136
max_download_threads=max_download_threads,
137137
ssl_options=ssl_options,
138138
statement_id=statement_id,
139-
chunk_id=0,
140139
schema_bytes=None,
141140
lz4_compressed=lz4_compressed,
142141
description=description,
143-
session_id_hex=None, # TODO: fix this argument when telemetry is implemented in SEA
142+
# TODO: fix these arguments when telemetry is implemented in SEA
143+
session_id_hex=None,
144+
chunk_id=0,
144145
)
145146

146147
self._sea_client = sea_client

src/databricks/sql/telemetry/latency_logger.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,9 @@ def log_latency(statement_type: StatementType = StatementType.NONE):
149149
- Creates a SqlExecutionEvent with execution details
150150
- Sends the telemetry data asynchronously via TelemetryClient
151151
152+
Args:
153+
statement_type (StatementType): The type of SQL statement being executed.
154+
152155
Usage:
153156
@log_latency(StatementType.QUERY)
154157
def execute(self, query):

0 commit comments

Comments
 (0)