diff --git a/machine-learning/app/test_main.py b/machine-learning/app/test_main.py index 8d63d63c76..a7648c5430 100644 --- a/machine-learning/app/test_main.py +++ b/machine-learning/app/test_main.py @@ -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(