Skip to content

Commit b97b9be

Browse files
CustomHttpClient -> SeaHttpClient
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 35ac6d0 commit b97b9be

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/databricks/sql/backend/sea_backend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from databricks.sql.backend.databricks_client import DatabricksClient
99
from databricks.sql.backend.types import SessionId, CommandId, CommandState, BackendType
1010
from databricks.sql.exc import Error, NotSupportedError, ServerOperationError
11-
from databricks.sql.backend.utils.http_client import CustomHttpClient
11+
from databricks.sql.backend.utils.http_client import SeaHttpClient
1212
from databricks.sql.thrift_api.TCLIService import ttypes
1313
from databricks.sql.types import SSLOptions
1414

@@ -77,7 +77,7 @@ def __init__(
7777
self.warehouse_id = self._extract_warehouse_id(http_path)
7878

7979
# Initialize HTTP client
80-
self.http_client = CustomHttpClient(
80+
self.http_client = SeaHttpClient(
8181
server_hostname=server_hostname,
8282
port=port,
8383
http_path=http_path,

tests/unit/test_sea_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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.CustomHttpClient"
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)