lint fix

Author: madhav-db

src/databricks/sql/auth/auth.py

@@ -13,7 +13,7 @@
 
 def get_auth_provider(cfg: ClientContext, http_client):
     # Determine the base auth provider
-    base_provider = None
+    base_provider: Optional[AuthProvider] = None
 
     if cfg.credentials_provider:
         base_provider = ExternalAuthProvider(cfg.credentials_provider)

src/databricks/sql/auth/token_federation.py

@@ -52,7 +52,7 @@ def __init__(
         self.identity_federation_client_id = identity_federation_client_id
 
         self._cached_token: Optional[Token] = None
-        self._external_headers = {}
+        self._external_headers: Dict[str, str] = {}
 
     def add_headers(self, request_headers: Dict[str, str]):
         """Add authentication headers to the request."""