Update session handler to retry on rate limit error

reichlab · Aug 26, 2024 · 56b1c71 · 56b1c71
1 parent 22ce19d
commit 56b1c71
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/reichlab_repo_utils/util/session.py b/src/reichlab_repo_utils/util/session.py
@@ -19,9 +19,9 @@ def get_session(token: str) -> requests.Session:
     # https://urllib3.readthedocs.io/en/latest/reference/urllib3.util.html#urllib3.util.retry.Retry
     retries = Retry(
         total=5,
-        allowed_methods=frozenset(["GET", "POST"]),
+        allowed_methods=frozenset(["GET", "POST", "PATCH"]),
         backoff_factor=1,
-        status_forcelist=[500, 502, 503, 504],
+        status_forcelist=[429, 500, 502, 503, 504],
     )
     session.mount("https://", HTTPAdapter(max_retries=retries))
     session.headers.update(headers)