fixes

saishreeeee · saishreeeee · commit f90cd995627c · 2025-07-29T16:24:13.000+05:30
Signed-off-by: Sai Shree Pradhan &lt;saishree.pradhan@databricks.com&gt;
diff --git a/src/databricks/sql/telemetry/telemetry_client.py b/src/databricks/sql/telemetry/telemetry_client.py
@@ -495,6 +495,7 @@ def connection_failure_log(
             session_id_hex=UNAUTH_DUMMY_SESSION_ID,
             auth_provider=None,
             host_url=host_url,
+            batch_size=TelemetryClientFactory.DEFAULT_BATCH_SIZE,
         )
 
         telemetry_client = TelemetryClientFactory.get_telemetry_client(
diff --git a/tests/unit/test_telemetry_retry.py b/tests/unit/test_telemetry_retry.py
@@ -47,6 +47,7 @@ def get_client(self, session_id, num_retries=3):
             session_id_hex=session_id,
             auth_provider=None,
             host_url="test.databricks.com",
+            batch_size=TelemetryClientFactory.DEFAULT_BATCH_SIZE
         )
         client = TelemetryClientFactory.get_telemetry_client(session_id)
         

Original file line number	Diff line number	Diff line change
`@@ -495,6 +495,7 @@ def connection_failure_log(`
`495`	`495`	`session_id_hex=UNAUTH_DUMMY_SESSION_ID,`
`496`	`496`	`auth_provider=None,`
`497`	`497`	`host_url=host_url,`
	`498`	`+ batch_size=TelemetryClientFactory.DEFAULT_BATCH_SIZE,`
`498`	`499`	`)`
`499`	`500`
`500`	`501`	`telemetry_client = TelemetryClientFactory.get_telemetry_client(`
Original file line number	Diff line number	Diff line change
`@@ -47,6 +47,7 @@ def get_client(self, session_id, num_retries=3):`
`47`	`47`	`session_id_hex=session_id,`
`48`	`48`	`auth_provider=None,`
`49`	`49`	`host_url="test.databricks.com",`
	`50`	`+ batch_size=TelemetryClientFactory.DEFAULT_BATCH_SIZE`
`50`	`51`	`)`
`51`	`52`	`client = TelemetryClientFactory.get_telemetry_client(session_id)`
`52`	`53`