Skip to content

Commit 1ceb658

Browse files
committed
changed order of args in telemetry client
Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>
1 parent 9887f9d commit 1ceb658

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

src/databricks/sql/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -302,9 +302,9 @@ def read(self) -> Optional[OAuthToken]:
302302
telemetry_client.initialize(
303303
host=self.host,
304304
connection_uuid=self.get_session_id_hex(),
305+
batch_size=telemetry_batch_size,
305306
auth_provider=auth_provider,
306307
is_authenticated=True, # TODO: Add authentication logic later
307-
batch_size=telemetry_batch_size,
308308
user_agent=useragent_header,
309309
)
310310

src/databricks/sql/telemetry/telemetry_client.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,9 @@ def __init__(
2727
self,
2828
host,
2929
connection_uuid,
30+
batch_size,
3031
auth_provider=None,
3132
is_authenticated=False,
32-
batch_size=200,
3333
user_agent=None,
3434
):
3535
self.host_url = host
@@ -195,19 +195,19 @@ def initialize(
195195
self,
196196
host,
197197
connection_uuid,
198-
auth_provider,
199-
is_authenticated,
200198
batch_size,
201-
user_agent,
199+
auth_provider=None,
200+
is_authenticated=False,
201+
user_agent=None,
202202
):
203203
"""Initialize a telemetry client for a specific connection"""
204204
if connection_uuid not in self._clients:
205205
self._clients[connection_uuid] = TelemetryClient(
206206
host=host,
207207
connection_uuid=connection_uuid,
208+
batch_size=batch_size,
208209
auth_provider=auth_provider,
209210
is_authenticated=is_authenticated,
210-
batch_size=batch_size,
211211
user_agent=user_agent,
212212
)
213213

0 commit comments

Comments
 (0)