From e6e4aa31adaad331a12a6f30d64fa12464d8660a Mon Sep 17 00:00:00 2001 From: Eu Jing Chua Date: Tue, 11 Feb 2025 22:59:40 +0000 Subject: [PATCH] Add 503 status code to retry --- .../services/remote/azure/azure_deployment_services.py | 6 +++++- 1 file changed, 5 insertions(+), 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 index 18b093ea93..1663f49d2b 100644 --- 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