fix in telemetry extractor

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

src/databricks/sql/telemetry/latency_logger.py

@@ -66,7 +66,7 @@ def get_retry_count(self) -> int:
             return len(self.thrift_backend.retry_policy.history)
         return 0
 
-    def get_statement_type(self: str) -> StatementType:
+    def get_statement_type(self) -> StatementType:
         # TODO: Implement this
         return StatementType.SQL
 
@@ -92,7 +92,7 @@ def get_execution_result(self) -> ExecutionResultFormat:
             return ExecutionResultFormat.INLINE_ARROW
         return ExecutionResultFormat.FORMAT_UNSPECIFIED
 
-    def get_statement_type(self: str) -> StatementType:
+    def get_statement_type(self) -> StatementType:
         # TODO: Implement this
         return StatementType.SQL
 

src/databricks/sql/telemetry/telemetry_client.py

@@ -400,6 +400,7 @@ def initialize_telemetry_client(
     ):
         """Initialize a telemetry client for a specific connection if telemetry is enabled"""
         try:
+
             with TelemetryClientFactory._lock:
                 TelemetryClientFactory._initialize()