Skip to content

Commit f559c79

Browse files
move sea stuff into sea/ dir
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 15a8efc commit f559c79

File tree

10 files changed

+11
-11
lines changed

10 files changed

+11
-11
lines changed
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,11 @@
1010
from databricks.sql.backend.databricks_client import DatabricksClient
1111
from databricks.sql.backend.types import SessionId, CommandId, CommandState, BackendType
1212
from databricks.sql.exc import Error, NotSupportedError, ServerOperationError
13-
from databricks.sql.backend.utils.http_client import CustomHttpClient
13+
from databricks.sql.backend.sea.utils.http_client import CustomHttpClient
1414
from databricks.sql.thrift_api.TCLIService import ttypes
1515
from databricks.sql.types import SSLOptions
1616

17-
from databricks.sql.backend.models import (
17+
from databricks.sql.backend.sea.models import (
1818
ExecuteStatementRequest,
1919
GetStatementRequest,
2020
CancelStatementRequest,

src/databricks/sql/backend/models/__init__.py renamed to src/databricks/sql/backend/sea/models/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
This package contains data models for SEA API requests and responses.
55
"""
66

7-
from databricks.sql.backend.models.base import (
7+
from databricks.sql.backend.sea.models.base import (
88
ServiceError,
99
StatementStatus,
1010
ExternalLink,
@@ -13,7 +13,7 @@
1313
ResultManifest,
1414
)
1515

16-
from databricks.sql.backend.models.requests import (
16+
from databricks.sql.backend.sea.models.requests import (
1717
StatementParameter,
1818
ExecuteStatementRequest,
1919
GetStatementRequest,
@@ -23,7 +23,7 @@
2323
DeleteSessionRequest,
2424
)
2525

26-
from databricks.sql.backend.models.responses import (
26+
from databricks.sql.backend.sea.models.responses import (
2727
ExecuteStatementResponse,
2828
GetStatementResponse,
2929
CreateSessionResponse,
File renamed without changes.
File renamed without changes.

src/databricks/sql/backend/models/responses.py renamed to src/databricks/sql/backend/sea/models/responses.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from dataclasses import dataclass, field
99

1010
from databricks.sql.backend.types import CommandState
11-
from databricks.sql.backend.models.base import (
11+
from databricks.sql.backend.sea.models.base import (
1212
StatementStatus,
1313
ResultManifest,
1414
ResultData,
File renamed without changes.

src/databricks/sql/backend/sea_result_set.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from databricks.sql.backend.types import CommandId, CommandState
1515
from databricks.sql.exc import Error
1616

17-
from databricks.sql.backend.models import (
17+
from databricks.sql.backend.sea.models import (
1818
StatementStatus,
1919
ResultManifest,
2020
ResultData,

src/databricks/sql/session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from databricks.sql import __version__
99
from databricks.sql import USER_AGENT_NAME
1010
from databricks.sql.backend.thrift_backend import ThriftDatabricksClient
11-
from databricks.sql.backend.sea_backend import SeaDatabricksClient
11+
from databricks.sql.backend.sea.backend import SeaDatabricksClient
1212
from databricks.sql.backend.databricks_client import DatabricksClient
1313
from databricks.sql.backend.types import SessionId, BackendType
1414

tests/unit/test_sea_backend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
import pytest
1010
from unittest.mock import patch, MagicMock, Mock
1111

12-
from databricks.sql.backend.sea_backend import SeaDatabricksClient
12+
from databricks.sql.backend.sea.backend import SeaDatabricksClient
1313
from databricks.sql.backend.sea_result_set import SeaResultSet
1414
from databricks.sql.backend.types import SessionId, CommandId, CommandState, BackendType
1515
from databricks.sql.types import SSLOptions
@@ -24,7 +24,7 @@ class TestSeaBackend:
2424
def mock_http_client(self):
2525
"""Create a mock HTTP client."""
2626
with patch(
27-
"databricks.sql.backend.sea_backend.CustomHttpClient"
27+
"databricks.sql.backend.sea.backend.CustomHttpClient"
2828
) as mock_client_class:
2929
mock_client = mock_client_class.return_value
3030
yield mock_client

tests/unit/test_sea_result_set.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
from databricks.sql.backend.sea_result_set import SeaResultSet
1414
from databricks.sql.backend.types import CommandState
15-
from databricks.sql.backend.models import (
15+
from databricks.sql.backend.sea.models import (
1616
StatementStatus,
1717
ResultManifest,
1818
ResultData,

0 commit comments

Comments
 (0)