From 4d0cd2e981ee6629aa6a52195bc0218b360c8ef4 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 14 Feb 2023 19:01:14 -0600 Subject: [PATCH] apply lint --- api/onnx_web/params.py | 3 ++- api/onnx_web/server/device_pool.py | 6 +++++- 2 files changed, 7 insertions(+), 2 deletions(-) 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(