apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): only consume one retry per error

This commit is contained in:
Sean Sube 2023-11-05 19:23:42 -06:00
parent df8b4abc77
commit 911f87f7ec
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/chain/base.py
View File

@ -251,11 +251,11 @@ class ChainPipeline:
break break
except Exception: except Exception:
logger.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() server.cache.clear()
run_gc([worker.get_device()]) run_gc([worker.get_device()])
worker.retries = worker.retries - (i + 1) worker.retries = worker.retries - 1
if worker.retries <= 0: if worker.retries <= 0:
raise RetryException("exhausted retries on stage") raise RetryException("exhausted retries on stage")