fix types and ignored pattern

2026-02-12 20:08:25 +03:00 · 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/models/base.py
+++ b/machine-learning/app/models/base.py
@@ -69,6 +69,7 @@ class InferenceModel(ABC):

    def _download(self) -> None:
        ignored_patterns: dict[ModelFormat, list[str]] = {
+            ModelFormat.ONNX: ["*.armnn", "*.rknn"],
            ModelFormat.ARMNN: ["*.rknn"],
            ModelFormat.RKNN: ["*.armnn"],
        }