formatting (black)

Signed-off-by: varun-edachali-dbx <[email protected]>

Author: varun-edachali-dbx

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

@@ -636,7 +636,9 @@ def get_execution_result(
             arraysize=cursor.arraysize,
         )
 
-    def get_chunk_links(self, statement_id: str, chunk_index: int) -> List[ExternalLink]:
+    def get_chunk_links(
+        self, statement_id: str, chunk_index: int
+    ) -> List[ExternalLink]:
         """
         Get links for chunks starting from the specified index.
         Args:

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

@@ -24,7 +24,12 @@
 from databricks.sql.exc import ProgrammingError
 from databricks.sql.thrift_api.TCLIService.ttypes import TSparkArrowResultLink
 from databricks.sql.types import SSLOptions
-from databricks.sql.utils import ArrowQueue, CloudFetchQueue, ResultSetQueue, create_arrow_table_from_arrow_file
+from databricks.sql.utils import (
+    ArrowQueue,
+    CloudFetchQueue,
+    ResultSetQueue,
+    create_arrow_table_from_arrow_file,
+)
 
 import logging
 
@@ -63,13 +68,15 @@ def build_queue(
             # INLINE disposition with JSON_ARRAY format
             return JsonQueue(result_data.data)
         elif manifest.format == ResultFormat.ARROW_STREAM.value:
-            if result_data.attachment is not None: 
+            if result_data.attachment is not None:
                 arrow_file = (
                     lz4.frame.decompress(result_data.attachment)
                     if lz4_compressed
                     else result_data.attachment
                 )
-                arrow_table = create_arrow_table_from_arrow_file(arrow_file, description)
+                arrow_table = create_arrow_table_from_arrow_file(
+                    arrow_file, description
+                )
                 return ArrowQueue(arrow_table, manifest.total_row_count)
 
             # EXTERNAL_LINKS disposition
@@ -207,7 +214,9 @@ def _progress_chunk_link(self):
         self._current_chunk_link = self._get_chunk_link(next_chunk_index)
         if not self._current_chunk_link:
             logger.error(
-                "SeaCloudFetchQueue: unable to retrieve link for chunk {}".format(next_chunk_index)
+                "SeaCloudFetchQueue: unable to retrieve link for chunk {}".format(
+                    next_chunk_index
+                )
             )
             return None