formatting

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

Author: saishreeeee

src/databricks/sql/telemetry/enums.py

@@ -40,4 +40,4 @@ class StatementType(Enum):
     QUERY = "query"
     SQL = "sql"
     UPDATE = "update"
-    METADATA = "metadata"
+    METADATA = "metadata"

src/databricks/sql/telemetry/telemetry_endpoint_models.py

@@ -20,4 +20,4 @@ class TelemetryResponse:
     numProtoSuccess: int
 
     def to_json(self):
-        return json.dumps(asdict(self))
+        return json.dumps(asdict(self))

src/databricks/sql/telemetry/telemetry_event.py

@@ -1,6 +1,13 @@
 import json
 from dataclasses import dataclass, asdict
-from databricks.sql.telemetry.enums import AuthMech, AuthFlow, DatabricksClientType, DriverVolumeOperationType, StatementType, ExecutionResultFormat
+from databricks.sql.telemetry.enums import (
+    AuthMech,
+    AuthFlow,
+    DatabricksClientType,
+    DriverVolumeOperationType,
+    StatementType,
+    ExecutionResultFormat,
+)
 
 
 @dataclass
@@ -51,7 +58,7 @@ class DriverConnectionParameters:
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
 
 @dataclass
 class DriverSystemConfiguration:
@@ -85,7 +92,7 @@ class DriverSystemConfiguration:
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
 
 @dataclass
 class DriverVolumeOperation:
@@ -101,7 +108,7 @@ class DriverVolumeOperation:
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
 
 @dataclass
 class DriverErrorInfo:
@@ -141,7 +148,7 @@ class SqlExecutionEvent:
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
 
 @dataclass
 class TelemetryEvent:
@@ -156,4 +163,4 @@ class TelemetryEvent:
     operation_latency_ms: int
 
     def to_json(self):
-        return json.dumps(asdict(self))
+        return json.dumps(asdict(self))

src/databricks/sql/telemetry/telemetry_frontend_log.py

@@ -2,6 +2,7 @@
 from dataclasses import dataclass, asdict
 from databricks.sql.telemetry.telemetry_event import TelemetryEvent
 
+
 @dataclass
 class TelemetryClientContext:
     timestamp_millis: int
@@ -15,15 +16,15 @@ class TelemetryClientContext:
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
 
 @dataclass
 class FrontendLogContext:
     client_context: TelemetryClientContext
 
     def to_json(self):
         return json.dumps(asdict(self))
-    
+
 
 @dataclass
 class FrontendLogEntry: