fix types and ignored pattern

2025-01-18 17:17:48 +08:00
parent 58f1cc92d7
commit 32f3707e52
3 changed files with 5 additions and 3 deletions
--- a/machine-learning/app/test_main.py
+++ b/machine-learning/app/test_main.py
@@ -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=runInference
+            rknnModel=Path(str(model_path).replace("model", "rk3566")).as_posix(), tpes=tpe, func=runInference
        )

        info.assert_has_calls(