rename sea.backend to sea.client for clarity

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

src/databricks/sql/backend/sea/backend.py src/databricks/sql/backend/sea/client.pysrc/databricks/sql/backend/sea/backend.py renamed to src/databricks/sql/backend/sea/client.py

@@ -3,7 +3,7 @@
 from abc import ABC
 from typing import List, Optional, Tuple
 
-from databricks.sql.backend.sea.backend import SeaDatabricksClient
+from databricks.sql.backend.sea.client import SeaDatabricksClient
 from databricks.sql.backend.sea.models.base import ResultData, ResultManifest
 from databricks.sql.backend.sea.utils.constants import ResultFormat
 from databricks.sql.exc import ProgrammingError

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

@@ -4,7 +4,7 @@
 
 import logging
 
-from databricks.sql.backend.sea.backend import SeaDatabricksClient
+from databricks.sql.backend.sea.client import SeaDatabricksClient
 from databricks.sql.backend.sea.models.base import ResultData, ResultManifest
 from databricks.sql.backend.sea.utils.conversion import SqlTypeConverter
 

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

@@ -71,7 +71,7 @@ def _filter_sea_result_set(
 
         result_data = ResultData(data=filtered_rows, external_links=None)
 
-        from databricks.sql.backend.sea.backend import SeaDatabricksClient
+        from databricks.sql.backend.sea.client import SeaDatabricksClient
         from databricks.sql.backend.sea.result_set import SeaResultSet
 
         # Create a new SeaResultSet with the filtered data

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

@@ -8,7 +8,7 @@
 from databricks.sql import __version__
 from databricks.sql import USER_AGENT_NAME
 from databricks.sql.backend.thrift_backend import ThriftDatabricksClient
-from databricks.sql.backend.sea.backend import SeaDatabricksClient
+from databricks.sql.backend.sea.client import SeaDatabricksClient
 from databricks.sql.backend.databricks_client import DatabricksClient
 from databricks.sql.backend.types import SessionId, BackendType
 

src/databricks/sql/session.py

@@ -13,7 +13,7 @@
 
 import lz4.frame
 
-from databricks.sql.backend.sea.backend import SeaDatabricksClient
+from databricks.sql.backend.sea.client import SeaDatabricksClient
 from databricks.sql.backend.sea.models.base import ResultData, ResultManifest
 
 try:

src/databricks/sql/utils.py

@@ -8,7 +8,7 @@
 import pytest
 from unittest.mock import patch, MagicMock, Mock
 
-from databricks.sql.backend.sea.backend import (
+from databricks.sql.backend.sea.client import (
     SeaDatabricksClient,
     _filter_session_configuration,
 )
@@ -31,7 +31,7 @@ class TestSeaBackend:
     def mock_http_client(self):
         """Create a mock HTTP client."""
         with patch(
-            "databricks.sql.backend.sea.backend.SeaHttpClient"
+            "databricks.sql.backend.sea.client.SeaHttpClient"
         ) as mock_client_class:
             mock_client = mock_client_class.return_value
             yield mock_client

tests/unit/test_sea_backend.py

@@ -26,7 +26,7 @@ def mock_connection(self):
 
         # Mock the session.backend to return a SeaDatabricksClient
         mock_session = Mock()
-        from databricks.sql.backend.sea.backend import SeaDatabricksClient
+        from databricks.sql.backend.sea.client import SeaDatabricksClient
 
         mock_backend = Mock(spec=SeaDatabricksClient)
         mock_backend.max_download_threads = 10