Apply black formatting to auth files

Author: madhav-db

src/databricks/sql/auth/auth.py

@@ -52,22 +52,22 @@ def __init__(
 def get_auth_provider(cfg: ClientContext):
     """
     Get an appropriate auth provider based on the provided configuration.
-    
+
     Token Federation Support:
     -----------------------
     Currently, token federation is implemented as a separate auth type, but the goal is to
     refactor it as a feature that can work with any auth type. The current implementation
     is maintained for backward compatibility while the refactoring is planned.
-    
+
     Future refactoring will introduce a `use_token_federation` flag that can be combined
     with any auth type to enable token federation.
-    
+
     Args:
         cfg: The client context containing configuration parameters
-        
+
     Returns:
         An appropriate AuthProvider instance
-        
+
     Raises:
         RuntimeError: If no valid authentication settings are provided
     """
@@ -163,22 +163,22 @@ def get_client_id_and_redirect_port(use_azure_auth: bool):
 def get_python_sql_connector_auth_provider(hostname: str, **kwargs):
     """
     Get an auth provider for the Python SQL connector.
-    
+
     This function is the main entry point for authentication in the SQL connector.
     It processes the parameters and creates an appropriate auth provider.
-    
+
     TODO: Future refactoring needed:
     1. Add a use_token_federation flag that can be combined with any auth type
     2. Remove TOKEN_FEDERATION as an auth_type while maintaining backward compatibility
     3. Create a token federation wrapper that can wrap any existing auth provider
-    
+
     Args:
         hostname: The Databricks server hostname
         **kwargs: Additional configuration parameters
-        
+
     Returns:
         An appropriate AuthProvider instance
-        
+
     Raises:
         ValueError: If username/password authentication is attempted (no longer supported)
     """

src/databricks/sql/auth/token_federation.py

@@ -150,7 +150,9 @@ def get_headers() -> Dict[str, str]:
                 else:
                     # Token is from a different host, need to exchange
                     logger.debug("Token from different host, attempting exchange")
-                    return self._try_token_exchange_or_fallback(access_token, token_type)
+                    return self._try_token_exchange_or_fallback(
+                        access_token, token_type
+                    )
             except Exception as e:
                 logger.error(f"Error processing token: {str(e)}")
                 # Fall back to original headers in case of error
@@ -324,13 +326,13 @@ def _send_token_exchange_request(
     ) -> Dict[str, Any]:
         """
         Send the token exchange request to the token endpoint.
-        
+
         Args:
             token_exchange_data: The data to send in the request
-            
+
         Returns:
             Dict[str, Any]: The parsed JSON response
-            
+
         Raises:
             Exception: If the request fails
         """