prevent circular imports

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

Author: varun-edachali-dbx

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

@@ -1,7 +1,7 @@
 import logging
 import time
 import re
-from typing import Any, Dict, Tuple, List, Optional, Union, TYPE_CHECKING, Set
+from typing import Any, Dict, Tuple, List, Optional, Union, TYPE_CHECKING, Set, cast
 
 from databricks.sql.backend.sea.models.base import ResultManifest
 from databricks.sql.backend.sea.utils.constants import (
@@ -12,7 +12,6 @@
     WaitTimeout,
     MetadataCommands,
 )
-from databricks.sql.result_set import SeaResultSet
 
 if TYPE_CHECKING:
     from databricks.sql.client import Cursor
@@ -723,13 +722,12 @@ def get_tables(
             enforce_embedded_schema_correctness=False,
         )
         assert result is not None, "execute_command returned None in synchronous mode"
-        assert isinstance(
-            result, SeaResultSet
-        ), "SEA backend execute_command returned a non-SeaResultSet"
 
         # Apply client-side filtering by table_types
         from databricks.sql.backend.sea.utils.filters import ResultSetFilter
+        from databricks.sql.result_set import SeaResultSet
 
+        result = cast(SeaResultSet, result)
         result = ResultSetFilter.filter_tables_by_type(result, table_types)
 
         return result

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

@@ -4,19 +4,22 @@
 This module provides filtering capabilities for result sets returned by different backends.
 """
 
+from __future__ import annotations
+
 import logging
 from typing import (
     List,
     Optional,
     Any,
     Callable,
     cast,
+    TYPE_CHECKING,
 )
 
-from databricks.sql.backend.sea.backend import SeaDatabricksClient
-from databricks.sql.backend.types import ExecuteResponse
+if TYPE_CHECKING:
+    from databricks.sql.result_set import SeaResultSet
 
-from databricks.sql.result_set import ResultSet, SeaResultSet
+from databricks.sql.backend.types import ExecuteResponse
 
 logger = logging.getLogger(__name__)
 
@@ -62,11 +65,11 @@ def _filter_sea_result_set(
         )
 
         # Create a new ResultData object with filtered data
-
         from databricks.sql.backend.sea.models.base import ResultData
 
         result_data = ResultData(data=filtered_rows, external_links=None)
 
+        from databricks.sql.backend.sea.backend import SeaDatabricksClient
         from databricks.sql.result_set import SeaResultSet
 
         # Create a new SeaResultSet with the filtered data