fixed linting issue

Signed-off-by: Nikhil Suri <[email protected]>

Author: nikhilsuri-db

src/databricks/sql/client.py

@@ -320,7 +320,9 @@ def read(self) -> Optional[OAuthToken]:
         )
 
         self.force_enable_telemetry = kwargs.get("force_enable_telemetry", False)
-        self.enable_telemetry = kwargs.get("enable_telemetry", True)  # Default to True for telemetry
+        self.enable_telemetry = kwargs.get(
+            "enable_telemetry", True
+        )  # Default to True for telemetry
         self.telemetry_enabled = TelemetryHelper.is_telemetry_enabled(self)
 
         TelemetryClientFactory.initialize_telemetry_client(

src/databricks/sql/common/feature_flag.py

@@ -107,8 +107,9 @@ def get_flag_value(self, name: str, default_value: Any) -> Any:
     def _refresh_flags(self):
         """Performs a synchronous network request to fetch and update flags."""
         import logging
+
         logger = logging.getLogger(__name__)
-        
+
         headers = {}
         try:
             # Authenticate the request
@@ -130,13 +131,17 @@ def _refresh_flags(self):
                 logger.info(f"Feature flags loaded: {self._flags}")
             else:
                 # On failure, initialize with an empty dictionary to prevent re-blocking.
-                logger.info(f"Feature flag fetch failed with status {response.status}, initializing empty flags")
+                logger.info(
+                    f"Feature flag fetch failed with status {response.status}, initializing empty flags"
+                )
                 if self._flags is None:
                     self._flags = {}
 
         except Exception as e:
             # On exception, initialize with an empty dictionary to prevent re-blocking.
-            logger.info(f"Feature flag fetch exception: {type(e).__name__}: {e}, initializing empty flags")
+            logger.info(
+                f"Feature flag fetch exception: {type(e).__name__}: {e}, initializing empty flags"
+            )
             if self._flags is None:
                 self._flags = {}
 

src/databricks/sql/telemetry/telemetry_client.py

@@ -114,7 +114,9 @@ def is_telemetry_enabled(connection: "Connection") -> bool:
             return True
 
         if connection.enable_telemetry:
-            logger.info(f"Telemetry: enable_telemetry=True, checking feature flag: {TelemetryHelper.TELEMETRY_FEATURE_FLAG_NAME}")
+            logger.info(
+                f"Telemetry: enable_telemetry=True, checking feature flag: {TelemetryHelper.TELEMETRY_FEATURE_FLAG_NAME}"
+            )
             context = FeatureFlagsContextFactory.get_instance(connection)
             flag_value = context.get_flag_value(
                 TelemetryHelper.TELEMETRY_FEATURE_FLAG_NAME, default_value=False
@@ -200,7 +202,9 @@ def _export_event(self, event):
         logger.info(f"Exporting telemetry event for connection {self._session_id_hex}")
         with self._lock:
             self._events_batch.append(event)
-        logger.info(f"Event added to batch, batch size now: {len(self._events_batch)}/{self._batch_size}")
+        logger.info(
+            f"Event added to batch, batch size now: {len(self._events_batch)}/{self._batch_size}"
+        )
         if len(self._events_batch) >= self._batch_size:
             logger.info(f"Batch size reached ({self._batch_size}), flushing events")
             self._flush()
@@ -485,17 +489,25 @@ def initialize_telemetry_client(
                             batch_size=batch_size,
                             client_context=client_context,
                         )
-                        logger.info(f"Created TelemetryClient for connection {session_id_hex}")
+                        logger.info(
+                            f"Created TelemetryClient for connection {session_id_hex}"
+                        )
                     else:
                         TelemetryClientFactory._clients[
                             session_id_hex
                         ] = NoopTelemetryClient()
-                        logger.info(f"Created NoopTelemetryClient for connection {session_id_hex}")
+                        logger.info(
+                            f"Created NoopTelemetryClient for connection {session_id_hex}"
+                        )
         except Exception as e:
-            logger.info(f"Failed to initialize telemetry client: {type(e).__name__}: {e}")
+            logger.info(
+                f"Failed to initialize telemetry client: {type(e).__name__}: {e}"
+            )
             # Fallback to NoopTelemetryClient to ensure connection doesn't fail
             TelemetryClientFactory._clients[session_id_hex] = NoopTelemetryClient()
-            logger.info(f"Fallback to NoopTelemetryClient for connection {session_id_hex}")
+            logger.info(
+                f"Fallback to NoopTelemetryClient for connection {session_id_hex}"
+            )
 
     @staticmethod
     def get_telemetry_client(session_id_hex):