apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-02-14 19:01:14 -06:00
parent 8a2a9174ba
commit 4d0cd2e981
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/params.py
View File

@ -1,7 +1,8 @@
from enum import IntEnum from enum import IntEnum
from onnxruntime import SessionOptions
from typing import Any, Dict, Literal, Optional, Tuple, Union from typing import Any, Dict, Literal, Optional, Tuple, Union
from onnxruntime import SessionOptions
class SizeChart(IntEnum): class SizeChart(IntEnum):
mini = 128 # small tile for very expensive models mini = 128 # small tile for very expensive models

6
api/onnx_web/server/device_pool.py
View File

@ -211,7 +211,11 @@ class DevicePoolExecutor:
# self.jobs[:] = [job for job in self.jobs if not job.future.done()] # self.jobs[:] = [job for job in self.jobs if not job.future.done()]
recent_count = len(self.recent) recent_count = len(self.recent)
if recent_count > self.recent_limit: 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 :] self.recent[:] = self.recent[-self.recent_limit :]
def submit( def submit(