diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py index 39990d89..b7d10903 100644 --- a/api/onnx_web/params.py +++ b/api/onnx_web/params.py @@ -1,7 +1,8 @@ from enum import IntEnum -from onnxruntime import SessionOptions from typing import Any, Dict, Literal, Optional, Tuple, Union +from onnxruntime import SessionOptions + class SizeChart(IntEnum): mini = 128 # small tile for very expensive models diff --git a/api/onnx_web/server/device_pool.py b/api/onnx_web/server/device_pool.py index c0b1c6a7..e11c920c 100644 --- a/api/onnx_web/server/device_pool.py +++ b/api/onnx_web/server/device_pool.py @@ -211,7 +211,11 @@ class DevicePoolExecutor: # self.jobs[:] = [job for job in self.jobs if not job.future.done()] recent_count = len(self.recent) if recent_count > self.recent_limit: - logger.debug("pruning %s of %s recent jobs", recent_count - self.recent_limit, recent_count) + logger.debug( + "pruning %s of %s recent jobs", + recent_count - self.recent_limit, + recent_count, + ) self.recent[:] = self.recent[-self.recent_limit :] def submit(