formatting (black + line gaps after multi-line pydocs)

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

Author: varun-edachali-dbx

examples/experimental/sea_connector_test.py

@@ -18,6 +18,7 @@ def test_sea_session():
     - DATABRICKS_HTTP_PATH: HTTP path for the SQL endpoint
     - DATABRICKS_TOKEN: Personal access token for authentication
     """
+
     server_hostname = os.environ.get("DATABRICKS_SERVER_HOSTNAME")
     http_path = os.environ.get("DATABRICKS_HTTP_PATH")
     access_token = os.environ.get("DATABRICKS_TOKEN")

src/databricks/sql/backend/sea_backend.py

@@ -55,8 +55,9 @@ def __init__(
             staging_allowed_local_path: Allowed local paths for staging operations
             **kwargs: Additional keyword arguments
         """
+
         logger.debug(
-            "SEADatabricksClient.__init__(server_hostname=%s, port=%s, http_path=%s)",
+            "SeaDatabricksClient.__init__(server_hostname=%s, port=%s, http_path=%s)",
             server_hostname,
             port,
             http_path,
@@ -97,6 +98,7 @@ def _extract_warehouse_id(self, http_path: str) -> str:
         Raises:
             Error: If the warehouse ID cannot be extracted from the path
         """
+
         path_parts = http_path.strip("/").split("/")
         warehouse_id = None
 
@@ -153,8 +155,9 @@ def open_session(
             Error: If the session configuration is invalid
             OperationalError: If there's an error establishing the session
         """
+
         logger.debug(
-            "SEADatabricksClient.open_session(session_configuration=%s, catalog=%s, schema=%s)",
+            "SeaDatabricksClient.open_session(session_configuration=%s, catalog=%s, schema=%s)",
             session_configuration,
             catalog,
             schema,
@@ -189,7 +192,8 @@ def close_session(self, session_id: SessionId) -> None:
             ValueError: If the session ID is invalid
             OperationalError: If there's an error closing the session
         """
-        logger.debug("SEADatabricksClient.close_session(session_id=%s)", session_id)
+
+        logger.debug("SeaDatabricksClient.close_session(session_id=%s)", session_id)
 
         if session_id.backend_type != BackendType.SEA:
             raise ValueError("Not a valid SEA session ID")

src/databricks/sql/backend/thrift_backend.py

@@ -53,7 +53,6 @@
 )
 from databricks.sql.types import SSLOptions
 from databricks.sql.backend.databricks_client import DatabricksClient
-from databricks.sql.result_set import ResultSet, ThriftResultSet
 
 logger = logging.getLogger(__name__)
 
@@ -1166,11 +1165,7 @@ def _handle_execute_response(self, resp, cursor):
             resp.directResults and resp.directResults.operationStatus,
         )
 
-        execute_response = self._results_message_to_execute_response(
-            resp, final_operation_state
-        )
-        execute_response = execute_response._replace(command_id=command_id)
-        return execute_response
+        return self._results_message_to_execute_response(resp, final_operation_state)
 
     def _handle_execute_response_async(self, resp, cursor):
         command_id = CommandId.from_thrift_handle(resp.operationHandle)

src/databricks/sql/backend/utils/http_client.py

@@ -40,6 +40,7 @@ def __init__(
             ssl_options: SSL configuration options
             **kwargs: Additional keyword arguments
         """
+
         self.server_hostname = server_hostname
         self.port = port
         self.http_path = http_path
@@ -103,6 +104,7 @@ def _make_request(
         Raises:
             RequestError: If the request fails
         """
+
         url = urljoin(self.base_url, path)
         headers = {**self.headers, **self._get_auth_headers()}