:= looks like walrus horizontally

Author: saishreeeee

src/databricks/sql/telemetry/telemetry_client.py

@@ -398,9 +398,8 @@ def close_telemetry_client(session_id_hex):
     """Remove the telemetry client for a specific connection"""
     global _initialized, _executor
     with _lock:
-        if session_id_hex in _clients:
+        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 = _clients.pop(session_id_hex, None)
             telemetry_client.close()
 
         # Shutdown executor if no more clients