redundant imports

Author: varun-edachali-dbx

src/databricks/sql/backend/sea_backend.py

@@ -1,6 +1,6 @@
 import logging
 import re
-from typing import Dict, Tuple, List, Optional, Union, TYPE_CHECKING
+from typing import Dict, Tuple, List, Optional, TYPE_CHECKING
 
 if TYPE_CHECKING:
     from databricks.sql.client import Cursor

src/databricks/sql/session.py

@@ -10,7 +10,7 @@
 from databricks.sql.backend.thrift_backend import ThriftDatabricksClient
 from databricks.sql.backend.sea_backend import SeaDatabricksClient
 from databricks.sql.backend.databricks_client import DatabricksClient
-from databricks.sql.backend.types import SessionId, BackendType
+from databricks.sql.backend.types import SessionId
 
 logger = logging.getLogger(__name__)
 

tests/unit/test_sea_backend.py

@@ -1,11 +1,11 @@
 import pytest
-from unittest.mock import patch, MagicMock, Mock
+from unittest.mock import patch, MagicMock
 
 from databricks.sql.backend.sea_backend import SeaDatabricksClient
 from databricks.sql.backend.types import SessionId, BackendType
 from databricks.sql.types import SSLOptions
 from databricks.sql.auth.authenticators import AuthProvider
-from databricks.sql.exc import Error, NotSupportedError
+from databricks.sql.exc import Error
 
 
 class TestSeaBackend:

tests/unit/test_session.py

@@ -1,12 +1,7 @@
 import unittest
-from unittest.mock import patch, MagicMock, Mock, PropertyMock
+from unittest.mock import patch, Mock
 import gc
 
-from databricks.sql.thrift_api.TCLIService.ttypes import (
-    TOpenSessionResp,
-    TSessionHandle,
-    THandleIdentifier,
-)
 from databricks.sql.backend.types import SessionId, BackendType
 
 import databricks.sql