Keep trying getting result even in case of HTTPError exception with s…

…tatus 520 in the deprecated package (#54)
Qiskit · Aug 29, 2024 · e228cd4 · e228cd4
1 parent e4f1e26
commit e228cd4
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/qiskit_transpiler_service/wrappers/base.py b/qiskit_transpiler_service/wrappers/base.py
@@ -104,6 +104,10 @@ def get_supported_backends(self):
         return resp
 
     def request_status(self, endpoint, task_id):
+        def _giveup(e):
+            # Only retry 520 errors
+            return e.response.status_code != 520
+
         @backoff.on_predicate(
             backoff.constant,
             lambda res: res.get("state") not in ["SUCCESS", "FAILURE"],
@@ -117,7 +121,9 @@ def request_status(self, endpoint, task_id):
                 requests.exceptions.Timeout,
                 requests.exceptions.ConnectionError,
                 requests.exceptions.JSONDecodeError,
+                requests.exceptions.HTTPError,
             ),
+            giveup=_giveup,
             max_time=self.timeout,
         )
         def _request_status(self, endpoint, task_id):