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.models.facial_recognition.recognition import FaceRecognizer
from app.sessions.ann import AnnSession from app.sessions.ann import AnnSession
from app.sessions.ort import OrtSession 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 .config import Settings, settings
from .models.base import InferenceModel from .models.base import InferenceModel
@ -355,7 +355,7 @@ class TestRknnSession:
RknnSession(model_path) RknnSession(model_path)
rknn_session.assert_called_once_with( 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( info.assert_has_calls(