remove redundant conversion.py

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

Author: varun-edachali-dbx

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

@@ -15,7 +15,6 @@
 
 if TYPE_CHECKING:
     from databricks.sql.client import Connection
-from databricks.sql.exc import ProgrammingError
 from databricks.sql.types import Row
 from databricks.sql.backend.sea.queue import JsonQueue, SeaResultSetQueueFactory
 from databricks.sql.backend.types import ExecuteResponse

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

@@ -1,16 +1,11 @@
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
-import json
-from typing import List, Optional, Any, Union, Tuple, TYPE_CHECKING
+from typing import List, Optional, Tuple, TYPE_CHECKING
 
 import logging
 import pandas
 
-from databricks.sql.backend.sea.backend import SeaDatabricksClient
-from databricks.sql.backend.sea.models.base import ResultData, ResultManifest
-from databricks.sql.backend.sea.conversion import SqlTypeConverter
-
 try:
     import pyarrow
 except ImportError:

src/databricks/sql/result_set.py

@@ -482,7 +482,7 @@ def test_command_management(
         )
 
         # Test get_query_state with invalid ID
-        with pytest.raises(ProgrammingError) as excinfo:
+        with pytest.raises(ValueError) as excinfo:
             sea_client.get_query_state(thrift_command_id)
         assert "Not a valid SEA command ID" in str(excinfo.value)
 

tests/unit/test_sea_backend.py

@@ -315,7 +315,6 @@ def test_iteration(self, result_set_with_data, sample_data):
         assert rows[0].col2 == 1
         assert rows[0].col3 is True
 
-
     def test_is_staging_operation(
         self, mock_connection, mock_sea_client, execute_response
     ):