Skip to content

Commit 4665083

Browse files
remove need for lazy load of SeaResultSet
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent e414d52 commit 4665083

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

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

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@
1919

2020
if TYPE_CHECKING:
2121
from databricks.sql.client import Cursor
22-
from databricks.sql.backend.sea.result_set import SeaResultSet
22+
23+
from databricks.sql.backend.sea.result_set import SeaResultSet
2324

2425
from databricks.sql.backend.databricks_client import DatabricksClient
2526
from databricks.sql.backend.types import (
@@ -375,9 +376,6 @@ def _response_to_result_set(
375376
Convert a SEA response to a SeaResultSet.
376377
"""
377378

378-
# Create and return a SeaResultSet
379-
from databricks.sql.backend.sea.result_set import SeaResultSet
380-
381379
execute_response = self._results_message_to_execute_response(response)
382380

383381
return SeaResultSet(

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
import logging
66

7-
from databricks.sql.backend.sea.backend import SeaDatabricksClient
87
from databricks.sql.backend.sea.models.base import ResultData, ResultManifest
98
from databricks.sql.backend.sea.utils.conversion import SqlTypeConverter
109

@@ -15,6 +14,7 @@
1514

1615
if TYPE_CHECKING:
1716
from databricks.sql.client import Connection
17+
from databricks.sql.backend.sea.backend import SeaDatabricksClient
1818
from databricks.sql.types import Row
1919
from databricks.sql.backend.sea.queue import JsonQueue, SeaResultSetQueueFactory
2020
from databricks.sql.backend.types import ExecuteResponse

0 commit comments

Comments
 (0)