old stuff without walrus seems to fail

Author: saishreeeee

src/databricks/sql/telemetry/telemetry_client.py

@@ -398,7 +398,9 @@ def close_telemetry_client(session_id_hex):
     """Remove the telemetry client for a specific connection"""
     global _initialized, _executor
     with _lock:
-        if (telemetry_client := _clients.pop(session_id_hex, None)) is not None:
+        # if (telemetry_client := _clients.pop(session_id_hex, None)) is not None:
+        if session_id_hex in _clients:
+            telemetry_client = _clients.pop(session_id_hex)
             logger.debug("Removing telemetry client for connection %s", session_id_hex)
             telemetry_client.close()