Skip to content

Commit 59b1330

Browse files
ensure SeaResultSet
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent a515d26 commit 59b1330

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
WaitTimeout,
1313
MetadataCommands,
1414
)
15+
from databricks.sql.result_set import SeaResultSet
1516

1617
if TYPE_CHECKING:
1718
from databricks.sql.client import Cursor
@@ -722,6 +723,9 @@ def get_tables(
722723
enforce_embedded_schema_correctness=False,
723724
)
724725
assert result is not None, "execute_command returned None in synchronous mode"
726+
assert isinstance(
727+
result, SeaResultSet
728+
), "SEA backend execute_command returned a non-SeaResultSet"
725729

726730
# Apply client-side filtering by table_types
727731
from databricks.sql.backend.sea.utils.filters import ResultSetFilter

0 commit comments

Comments
 (0)