fix(api): only consume one retry per error

ssube · Nov 6, 2023 · 911f87f · 911f87f
1 parent df8b4ab
commit 911f87f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/onnx_web/chain/base.py b/api/onnx_web/chain/base.py
@@ -251,11 +251,11 @@ def stage_tile(
                         break
                     except Exception:
                         logger.exception(
-                            "error while running stage pipeline, retry %s of 3", i
+                            "error while running stage pipeline, %s retries left", worker.retries
                         )
                         server.cache.clear()
                         run_gc([worker.get_device()])
-                        worker.retries = worker.retries - (i + 1)
+                        worker.retries = worker.retries - 1
 
                 if worker.retries <= 0:
                     raise RetryException("exhausted retries on stage")