Skip to content

Commit 28fcc80

Browse files
SNOW-2274410: Set no retries for WIF authentication (#2494)
1 parent eeef4b9 commit 28fcc80

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

src/snowflake/connector/auth/workload_identity.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,9 @@ def prepare(
8585
self.provider,
8686
self.entra_resource,
8787
self.token,
88-
session_manager=conn._session_manager.clone() if conn else None,
88+
session_manager=(
89+
conn._session_manager.clone(max_retries=0) if conn else None
90+
),
8991
)
9092

9193
def reauthenticate(self, **kwargs: typing.Any) -> dict[str, bool]:

src/snowflake/connector/wif_util.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,9 @@ def create_attestation(
303303
"""
304304
entra_resource = entra_resource or DEFAULT_ENTRA_SNOWFLAKE_RESOURCE
305305
session_manager = (
306-
session_manager.clone() if session_manager else SessionManager(use_pooling=True)
306+
session_manager.clone()
307+
if session_manager
308+
else SessionManager(use_pooling=True, max_retries=0)
307309
)
308310

309311
if provider == AttestationProvider.AWS:

0 commit comments

Comments
 (0)