manually do the walrussing

Author: saishreeeee

src/databricks/sql/telemetry/telemetry_client.py

@@ -399,8 +399,8 @@ def close_telemetry_client(session_id_hex):
     global _initialized, _executor
     with _lock:
         # 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)
+        telemetry_client = _clients.pop(session_id_hex, None)
+        if telemetry_client is not None:
             logger.debug("Removing telemetry client for connection %s", session_id_hex)
             telemetry_client.close()