From b3ae5d34ccb525614c6d0d2fa6231cdec171f157 Mon Sep 17 00:00:00 2001 From: yoni13 Date: Sat, 18 Jan 2025 16:54:18 +0800 Subject: [PATCH] fix typo in tests --- machine-learning/app/test_main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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(