fix(api): run garbage collection after job errors

ssube · Feb 10, 2023 · b3e4076 · b3e4076
1 parent 45b0916
commit b3e4076
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py
@@ -6,6 +6,7 @@
 from typing import Any, Callable, List, Optional, Tuple, Union
 
 from .params import DeviceParams
+from .utils import run_gc
 
 logger = getLogger(__name__)
 
@@ -197,6 +198,7 @@ def job_done(f: Future):
                     key,
                     format_exception(type(err), err, err.__traceback__),
                 )
+                run_gc()
 
         future.add_done_callback(job_done)