apply lint
This commit is contained in:
parent
8a2a9174ba
commit
4d0cd2e981
|
@ -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
|
||||||
|
|
|
@ -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(
|
||||||
|
|
Loading…
Reference in New Issue