diff --git a/src/snowflake/connector/auth/workload_identity.py b/src/snowflake/connector/auth/workload_identity.py index c52eaa2b22..7ab4c41d19 100644 --- a/src/snowflake/connector/auth/workload_identity.py +++ b/src/snowflake/connector/auth/workload_identity.py @@ -85,7 +85,9 @@ def prepare( self.provider, self.entra_resource, self.token, - session_manager=conn._session_manager.clone() if conn else None, + session_manager=( + conn._session_manager.clone(max_retries=0) if conn else None + ), ) def reauthenticate(self, **kwargs: typing.Any) -> dict[str, bool]: diff --git a/src/snowflake/connector/wif_util.py b/src/snowflake/connector/wif_util.py index 406ee12725..f01026b6b9 100644 --- a/src/snowflake/connector/wif_util.py +++ b/src/snowflake/connector/wif_util.py @@ -303,7 +303,9 @@ def create_attestation( """ entra_resource = entra_resource or DEFAULT_ENTRA_SNOWFLAKE_RESOURCE session_manager = ( - session_manager.clone() if session_manager else SessionManager(use_pooling=True) + session_manager.clone() + if session_manager + else SessionManager(use_pooling=True, max_retries=0) ) if provider == AttestationProvider.AWS: