fix typo in tests

This commit is contained in:
yoni13 2025-01-18 16:54:18 +08:00
parent 4e42fbc091
commit b3ae5d34cc

View File

@ -24,7 +24,7 @@ from app.models.facial_recognition.detection import FaceDetector
from app.models.facial_recognition.recognition import FaceRecognizer
from app.sessions.ann import AnnSession
from app.sessions.ort import OrtSession
from app.sessions.rknn import RknnSession, runInfrence
from app.sessions.rknn import RknnSession, runInference
from .config import Settings, settings
from .models.base import InferenceModel
@ -355,7 +355,7 @@ class TestRknnSession:
RknnSession(model_path)
rknn_session.assert_called_once_with(
rknnModel=Path(str(model_path).replace("model", "rk3566")).as_posix(), TPEs=tpe, func=runInfrence
rknnModel=Path(str(model_path).replace("model", "rk3566")).as_posix(), TPEs=tpe, func=runInference
)
info.assert_has_calls(