Skip to content

Commit 5b7e29f

Browse files
committed
Do not ignore retry_config parameter in AuthenticatedFlightClient
1 parent 1a9ecbb commit 5b7e29f

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

graphdatascience/arrow_client/authenticated_flight_client.py

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -102,16 +102,7 @@ def __init__(
102102
self._user_agent = user_agent
103103
self._retry_config = retry_config
104104
self._logger = logging.getLogger("gds_arrow_client")
105-
self._retry_config = RetryConfig(
106-
retry=retry_any(
107-
retry_if_exception_type(FlightTimedOutError),
108-
retry_if_exception_type(FlightUnavailableError),
109-
retry_if_exception_type(FlightInternalError),
110-
),
111-
stop=(stop_after_delay(10) | stop_after_attempt(5)),
112-
wait=wait_exponential(multiplier=1, min=1, max=10),
113-
)
114-
105+
self._retry_config = retry_config
115106
if auth:
116107
self._auth = auth
117108
self._auth_middleware = AuthMiddleware(auth)

0 commit comments

Comments
 (0)