fix type issues

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

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

@@ -511,6 +511,8 @@ def cancel_command(self, command_id: CommandId) -> None:
             raise ProgrammingError("Not a valid SEA command ID")
 
         sea_statement_id = command_id.to_sea_statement_id()
+        if sea_statement_id is None:
+            raise ProgrammingError("Not a valid SEA command ID")
 
         request = CancelStatementRequest(statement_id=sea_statement_id)
         self.http_client._make_request(
@@ -534,6 +536,8 @@ def close_command(self, command_id: CommandId) -> None:
             raise ProgrammingError("Not a valid SEA command ID")
 
         sea_statement_id = command_id.to_sea_statement_id()
+        if sea_statement_id is None:
+            raise ProgrammingError("Not a valid SEA command ID")
 
         request = CloseStatementRequest(statement_id=sea_statement_id)
         self.http_client._make_request(
@@ -560,6 +564,8 @@ def get_query_state(self, command_id: CommandId) -> CommandState:
             raise ValueError("Not a valid SEA command ID")
 
         sea_statement_id = command_id.to_sea_statement_id()
+        if sea_statement_id is None:
+            raise ProgrammingError("Not a valid SEA command ID")
 
         request = GetStatementRequest(statement_id=sea_statement_id)
         response_data = self.http_client._make_request(
@@ -595,6 +601,8 @@ def get_execution_result(
             raise ProgrammingError("Not a valid SEA command ID")
 
         sea_statement_id = command_id.to_sea_statement_id()
+        if sea_statement_id is None:
+            raise ProgrammingError("Not a valid SEA command ID")
 
         # Create the request model
         request = GetStatementRequest(statement_id=sea_statement_id)

src/databricks/sql/result_set.py

@@ -522,9 +522,6 @@ def _convert_json_types(self, rows: List[List]) -> List[List]:
         Convert string values to appropriate Python types based on column metadata.
         """
 
-        if not self.description or not rows:
-            return rows
-
         # JSON + INLINE gives us string values, so we convert them to appropriate
         #   types based on column metadata
         converted_rows = []
@@ -562,9 +559,6 @@ def _create_json_table(self, rows: List[List]) -> List[Row]:
             List of Row objects with named columns and converted values
         """
 
-        if not self.description or not rows:
-            return rows
-
         ResultRow = Row(*[col[0] for col in self.description])
         rows = self._convert_json_types(rows)
 

src/databricks/sql/utils.py

@@ -274,7 +274,7 @@ def __init__(
         start_row_offset: int = 0,
         result_links: Optional[List[TSparkArrowResultLink]] = None,
         lz4_compressed: bool = True,
-        description: List[Tuple] = [],
+        description: Optional[List[Tuple]] = None,
     ):
         """
         A queue-like wrapper over CloudFetch arrow batches.