@@ -150,9 +150,7 @@ def runtime_native_auth(cfg: "Config") -> Optional[CredentialsProvider]:
150150 # This import MUST be after the "DATABRICKS_RUNTIME_VERSION" check
151151 # above, so that we are not throwing import errors when not in
152152 # runtime and no config variables are set.
153- from databricks .sdk .runtime import (init_runtime_legacy_auth ,
154- init_runtime_native_auth ,
155- init_runtime_repl_auth )
153+ from databricks .sdk .runtime import init_runtime_legacy_auth , init_runtime_native_auth , init_runtime_repl_auth
156154
157155 for init in [
158156 init_runtime_native_auth ,
@@ -405,10 +403,11 @@ def token() -> oauth.Token:
405403
406404 return OAuthCredentialsProvider (refreshed_headers , token )
407405
406+
408407@oauth_credentials_strategy ("azdo-oidc" , ["host" , "client_id" ])
409408def azure_devops_oidc (cfg : "Config" ) -> Optional [CredentialsProvider ]:
410409 """
411- Azure DevOps OIDC authentication uses a Token Supplier to get a JWT Token
410+ Azure DevOps OIDC authentication uses a Token Supplier to get a JWT Token
412411 and exchanges it for a Databricks Token.
413412
414413 Supported in Azure DevOps pipelines with OIDC service connections.
@@ -456,7 +455,6 @@ def token() -> oauth.Token:
456455 return OAuthCredentialsProvider (refreshed_headers , token )
457456
458457
459-
460458@oauth_credentials_strategy ("github-oidc-azure" , ["host" , "azure_client_id" ])
461459def github_oidc_azure (cfg : "Config" ) -> Optional [CredentialsProvider ]:
462460 if "ACTIONS_ID_TOKEN_REQUEST_TOKEN" not in os .environ :
0 commit comments