File tree Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Expand file tree Collapse file tree 2 files changed +6
-2
lines changed Original file line number Diff line number Diff 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 ]:
Original file line number Diff line number Diff 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 :
You can’t perform that action at this time.
0 commit comments