Reformatted

Author: jprakash-db

src/databricks/sql/auth/thrift_http_client.py

@@ -192,13 +192,17 @@ def flush(self):
             timeout=self.__timeout,
             retries=self.retry_policy,
         )
-        
+
         # Get reply to flush the request
         self.code = self.__resp.status
         self.message = self.__resp.reason
         self.headers = self.__resp.headers
 
-        logger.info("HTTP Response with status code {}, message: {}".format(self.code, self.message))
+        logger.info(
+            "HTTP Response with status code {}, message: {}".format(
+                self.code, self.message
+            )
+        )
 
     @staticmethod
     def basic_proxy_auth_headers(proxy):

src/databricks/sql/thrift_backend.py

@@ -382,7 +382,9 @@ def attempt_request(attempt):
 
                 # We need to call type(response) here because thrift doesn't implement __name__ attributes for thrift responses
                 logger.debug(
-                    "Received response: {}(<REDACTED>) with status {}".format(type(response).__name__, getattr(response, "status", None))
+                    "Received response: {}(<REDACTED>) with status {}".format(
+                        type(response).__name__, getattr(response, "status", None)
+                    )
                 )
                 unsafe_logger.debug("Received response: {}".format(response))
                 return response