diff --git a/terrat_runner/requests_retry.py b/terrat_runner/requests_retry.py index c4109007..eea1f823 100644 --- a/terrat_runner/requests_retry.py +++ b/terrat_runner/requests_retry.py @@ -4,7 +4,7 @@ import retry - +TIMEOUT = 120 TRIES = 5 INITIAL_SLEEP = 1 BACKOFF = 1.5 @@ -39,12 +39,12 @@ def _wrap(f): def post(*args, **kwargs): - return _wrap(lambda: requests.post(*args, **kwargs)) + return _wrap(lambda: requests.post(*args, timeout=TIMEOUT, **kwargs)) def put(*args, **kwargs): - return _wrap(lambda: requests.put(*args, **kwargs)) + return _wrap(lambda: requests.put(*args, timeout=TIMEOUT, **kwargs)) def get(*args, **kwargs): - return _wrap(lambda: requests.get(*args, **kwargs)) + return _wrap(lambda: requests.get(*args, timeout=TIMEOUT, **kwargs))