ruff

machine-learning/app/models/base.py

@@ -8,11 +8,10 @@ from typing import Any, ClassVar
 from huggingface_hub import snapshot_download
 
 import ann.ann
+import rknn.rknnpool
 from app.sessions.ort import OrtSession
 from app.sessions.rknn import RknnSession
 
-import rknn.rknnpool
-
 from ..config import clean_name, log, settings
 from ..schemas import ModelFormat, ModelIdentity, ModelSession, ModelTask, ModelType
 from ..sessions.ann import AnnSession

machine-learning/app/sessions/rknn.py

@@ -6,9 +6,9 @@ from typing import Any
 import numpy as np
 import onnxruntime as ort
 from numpy.typing import NDArray
-from rknn.rknnpool import rknnPoolExecutor
 
 from app.schemas import SessionNode
+from rknn.rknnpool import rknnPoolExecutor
 
 from ..config import log
 

machine-learning/rknn/rknnpool.py

@@ -1,8 +1,8 @@
 # This code is from leafqycc/rknn-multi-threaded
 # Following Apache License 2.0
 
+from concurrent.futures import ThreadPoolExecutor
 from queue import Queue
-from concurrent.futures import ThreadPoolExecutor, as_completed
 
 supported_socs = ["rk3566", "rk3568", "rk3588"]