walrus again

Author: saishreeeee

src/databricks/sql/telemetry/telemetry_client.py

@@ -398,9 +398,7 @@ 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 session_id_hex in _clients:
-            telemetry_client = _clients.pop(session_id_hex)
+        if (telemetry_client := _clients.pop(session_id_hex, None)) is not None:
             logger.debug("Removing telemetry client for connection %s", session_id_hex)
             telemetry_client.close()