Add 503 status code to retry

microsoft · Feb 11, 2025 · e6e4aa3 · e6e4aa3
1 parent e94989e
commit e6e4aa3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mlos_bench/mlos_bench/services/remote/azure/azure_deployment_services.py b/mlos_bench/mlos_bench/services/remote/azure/azure_deployment_services.py
@@ -140,7 +140,11 @@ def _get_session(self, params: dict) -> requests.Session:
         session = requests.Session()
         session.mount(
             "https://",
-            HTTPAdapter(max_retries=Retry(total=total_retries, backoff_factor=backoff_factor)),
+            HTTPAdapter(
+                max_retries=Retry(
+                    total=total_retries, backoff_factor=backoff_factor, status_forcelist=[503]
+                )
+            ),
         )
         session.headers.update(self._get_headers())
         return session