Skip to content

Commit 4ff64ed

Browse files
move sea backend and models into separate sea/ dir
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent ed4931e commit 4ff64ed

File tree

6 files changed

+6
-6
lines changed

6 files changed

+6
-6
lines changed

src/databricks/sql/backend/sea_backend.py renamed to src/databricks/sql/backend/sea/backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from databricks.sql.thrift_api.TCLIService import ttypes
1313
from databricks.sql.types import SSLOptions
1414

15-
from databricks.sql.backend.models import (
15+
from databricks.sql.backend.sea.models import (
1616
CreateSessionRequest,
1717
DeleteSessionRequest,
1818
CreateSessionResponse,

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

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

7-
from databricks.sql.backend.models.requests import (
7+
from databricks.sql.backend.sea.models.requests import (
88
CreateSessionRequest,
99
DeleteSessionRequest,
1010
)
1111

12-
from databricks.sql.backend.models.responses import (
12+
from databricks.sql.backend.sea.models.responses import (
1313
CreateSessionResponse,
1414
)
1515

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
1414

tests/unit/test_sea_backend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import pytest
22
from unittest.mock import patch, MagicMock
33

4-
from databricks.sql.backend.sea_backend import SeaDatabricksClient
4+
from databricks.sql.backend.sea.backend import SeaDatabricksClient
55
from databricks.sql.backend.types import SessionId, BackendType
66
from databricks.sql.types import SSLOptions
77
from databricks.sql.auth.authenticators import AuthProvider
@@ -15,7 +15,7 @@ class TestSeaBackend:
1515
def mock_http_client(self):
1616
"""Create a mock HTTP client."""
1717
with patch(
18-
"databricks.sql.backend.sea_backend.SeaHttpClient"
18+
"databricks.sql.backend.sea.backend.SeaHttpClient"
1919
) as mock_client_class:
2020
mock_client = mock_client_class.return_value
2121
yield mock_client

0 commit comments

Comments
 (0)