fixed format

Signed-off-by: Sai Shree Pradhan <[email protected]>

Author: saishreeeee

src/databricks/sql/client.py

@@ -214,7 +214,9 @@ def read(self) -> Optional[OAuthToken]:
         # use_cloud_fetch
         # Enable use of cloud fetch to extract large query results in parallel via cloud storage
 
-        logger.debug(f"Connection.__init__(server_hostname={server_hostname}, http_path={http_path})")
+        logger.debug(
+            f"Connection.__init__(server_hostname={server_hostname}, http_path={http_path})"
+        )
 
         if access_token:
             access_token_kv = {"access_token": access_token}

src/databricks/sql/thrift_backend.py

@@ -131,7 +131,9 @@ def __init__(
         # max_download_threads
         #  Number of threads for handling cloud fetch downloads. Defaults to 10
 
-        logger.debug(f"ThriftBackend.__init__(server_hostname={server_hostname}, port={port}, http_path={http_path})")
+        logger.debug(
+            f"ThriftBackend.__init__(server_hostname={server_hostname}, port={port}, http_path={http_path})"
+        )
 
         port = port or 443
         if kwargs.get("_connection_uri"):
@@ -392,7 +394,7 @@ def attempt_request(attempt):
 
                 # TODO: don't use exception handling for GOS polling...
 
-                logger.debug(f"ThriftBackend.attempt_request: HTTPError: {err}")
+                logger.error(f"ThriftBackend.attempt_request: HTTPError: {err}")
 
                 gos_name = TCLIServiceClient.GetOperationStatus.__name__
                 if method.__name__ == gos_name:
@@ -408,7 +410,7 @@ def attempt_request(attempt):
                 else:
                     raise err
             except OSError as err:
-                logger.debug(f"ThriftBackend.attempt_request: OSError: {err}")
+                logger.error(f"ThriftBackend.attempt_request: OSError: {err}")
                 error = err
                 error_message = str(err)
                 # fmt: off
@@ -439,7 +441,7 @@ def attempt_request(attempt):
                     else:
                         logger.warning(log_string)
             except Exception as err:
-                logger.debug(f"ThriftBackend.attempt_request: Exception: {err}")
+                logger.error(f"ThriftBackend.attempt_request: Exception: {err}")
                 error = err
                 retry_delay = extract_retry_delay(attempt)
                 error_message = ThriftBackend._extract_error_message_from_headers(
@@ -894,7 +896,10 @@ def execute_command(
     ):
         assert session_handle is not None
 
-        logger.debug(f"ThriftBackend.execute_command(operation={operation}, session_handle={session_handle})")
+        logger.debug(
+            f"ThriftBackend.execute_command(operation={operation}, session_handle={session_handle})"
+        )
+
         spark_arrow_types = ttypes.TSparkArrowTypes(
             timestampAsArrow=self._use_arrow_native_timestamps,
             decimalAsArrow=self._use_arrow_native_decimals,