@@ -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 ,
@@ -406,10 +404,11 @@ def token() -> oauth.Token:
406404
407405 return OAuthCredentialsProvider (refreshed_headers , token )
408406
407+
409408@oauth_credentials_strategy ("azdo-oidc" , ["host" , "client_id" ])
410409def azure_devops_oidc (cfg : "Config" ) -> Optional [CredentialsProvider ]:
411410 """
412- Azure DevOps OIDC authentication uses a Token Supplier to get a JWT Token
411+ Azure DevOps OIDC authentication uses a Token Supplier to get a JWT Token
413412 and exchanges it for a Databricks Token.
414413
415414 Supported in Azure DevOps pipelines with OIDC service connections.
@@ -457,7 +456,6 @@ def token() -> oauth.Token:
457456 return OAuthCredentialsProvider (refreshed_headers , token )
458457
459458
460-
461459@oauth_credentials_strategy ("github-oidc-azure" , ["host" , "azure_client_id" ])
462460def github_oidc_azure (cfg : "Config" ) -> Optional [CredentialsProvider ]:
463461 if "ACTIONS_ID_TOKEN_REQUEST_TOKEN" not in os .environ :
0 commit comments