diff --git a/machine-learning/app/sessions/rknn.py b/machine-learning/app/sessions/rknn.py index c79e669632..90bea605ca 100644 --- a/machine-learning/app/sessions/rknn.py +++ b/machine-learning/app/sessions/rknn.py @@ -66,7 +66,7 @@ class RknnSession: input_feed: dict[str, NDArray[np.float32]] | dict[str, NDArray[np.int32]], run_options: Any = None, ) -> list[NDArray[np.float32]]: - input_data: list[NDArray[np.float32]] = [np.ascontiguousarray(v) for v in input_feed.values()] + input_data: list[NDArray[np.float32]] = [np.ascontiguousarray(v) for v in input_feed.values()] self.rknnpool.put(input_data) outputs: list[NDArray[np.float32]] = self.rknnpool.get() return outputs