diff --git a/tap_sailthru/client.py b/tap_sailthru/client.py index 063ff30..046a564 100644 --- a/tap_sailthru/client.py +++ b/tap_sailthru/client.py @@ -12,6 +12,8 @@ from requests import Session from requests.exceptions import Timeout from singer import get_logger, metrics +from requests.exceptions import ConnectionError +from http.client import RemoteDisconnected LOGGER = get_logger() @@ -323,7 +325,9 @@ def _build_request(self, endpoint, params, method): (SailthruClientError, SailthruServer5xxError, SailthruClientStatsNotReadyError, - Timeout), + Timeout, + ConnectionError, + RemoteDisconnected), max_tries=MAX_RETRIES, factor=2) def _make_request(self, url, payload, method):