simplify diff

Author: Varun0157

src/databricks/sql/backend/sea/backend.py

@@ -287,9 +287,9 @@ def close_session(self, session_id: SessionId) -> None:
 
         logger.debug("SeaDatabricksClient.close_session(session_id=%s)", session_id)
 
-        sea_session_id = session_id.to_sea_session_id()
-        if sea_session_id is None:
+        if session_id.backend_type != BackendType.SEA:
             raise ValueError("Not a valid SEA session ID")
+        sea_session_id = session_id.to_sea_session_id()
 
         request_data = DeleteSessionRequest(
             warehouse_id=self.warehouse_id,

src/databricks/sql/telemetry/telemetry_client.py

@@ -190,6 +190,7 @@ def __init__(
         self._session_id_hex = session_id_hex
         self._auth_provider = auth_provider
         self._user_agent = None
+        self._lock = threading.RLock()
         self._pending_futures = set()
 
         # OPTIMIZATION: Use lock-free Queue instead of list + lock
@@ -443,9 +444,9 @@ class TelemetryClientFactory:
     It uses a thread pool to handle asynchronous operations and a single flush thread for all clients.
     """
 
-    _clients: Dict[str, BaseTelemetryClient] = (
-        {}
-    )  # Map of session_id_hex -> BaseTelemetryClient
+    _clients: Dict[
+        str, BaseTelemetryClient
+    ] = {} # Map of session_id_hex -> BaseTelemetryClient
     _executor: Optional[ThreadPoolExecutor] = None
     _initialized: bool = False
     _lock = threading.RLock()  # Thread safety for factory operations
@@ -546,8 +547,9 @@ def initialize_telemetry_client(
                         session_id_hex,
                     )
                     if telemetry_enabled:
-                        TelemetryClientFactory._clients[session_id_hex] = (
-                            TelemetryClient(
+                        TelemetryClientFactory._clients[
+                            session_id_hex
+                        ] = TelemetryClient(
                                 telemetry_enabled=telemetry_enabled,
                                 session_id_hex=session_id_hex,
                                 auth_provider=auth_provider,
@@ -556,11 +558,10 @@ def initialize_telemetry_client(
                                 batch_size=batch_size,
                                 client_context=client_context,
                             )
-                        )
                     else:
-                        TelemetryClientFactory._clients[session_id_hex] = (
-                            NoopTelemetryClient()
-                        )
+                        TelemetryClientFactory._clients[
+                            session_id_hex
+                        ] = NoopTelemetryClient()
         except Exception as e:
             logger.debug("Failed to initialize telemetry client: %s", e)
             # Fallback to NoopTelemetryClient to ensure connection doesn't fail