Fixed the exception handler close() on _TelemetryClientHolder (databricks#723)

msrathore-db · web-flow · commit 9b4e5770ab21 · 2026-01-02T01:18:14.000+05:30
Fixed the exception handler calls close() on _TelemetryClientHolder objects instead of accessing the client inside them.
diff --git a/src/databricks/sql/telemetry/telemetry_client.py b/src/databricks/sql/telemetry/telemetry_client.py
@@ -542,8 +542,8 @@ def _handle_unhandled_exception(cls, exc_type, exc_value, exc_traceback):
         logger.debug("Handling unhandled exception: %s", exc_type.__name__)
 
         clients_to_close = list(cls._clients.values())
-        for client in clients_to_close:
-            client.close()
+        for holder in clients_to_close:
+            holder.client.close()
 
         # Call the original exception handler to maintain normal behavior
         if cls._original_excepthook:
