Skip to content

Commit f74836d

Browse files
fix type issues
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 87d73da commit f74836d

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

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

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,8 @@
2727
class SeaResultSet(ResultSet):
2828
"""ResultSet implementation for SEA backend."""
2929

30+
backend: SeaDatabricksClient
31+
3032
def __init__(
3133
self,
3234
connection: Connection,
@@ -83,10 +85,6 @@ def __init__(
8385
arrow_schema_bytes=execute_response.arrow_schema_bytes,
8486
)
8587

86-
assert isinstance(
87-
self.backend, SeaDatabricksClient
88-
), "SeaResultSet must be used with SeaDatabricksClient"
89-
9088
def _convert_complex_types_to_string(
9189
self, rows: "pyarrow.Table"
9290
) -> "pyarrow.Table":

0 commit comments

Comments
 (0)