From b3e4076775e746e665d9d07384bb8efd7da971a5 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 9 Feb 2023 22:04:33 -0600 Subject: [PATCH] fix(api): run garbage collection after job errors --- api/onnx_web/device_pool.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index 6ece8322..42faddee 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -6,6 +6,7 @@ from traceback import format_exception 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 @@ class DevicePoolExecutor: key, format_exception(type(err), err, err.__traceback__), ) + run_gc() future.add_done_callback(job_done)