From 000eb5f78469724d56b371039a68b6f3dc05ddfd Mon Sep 17 00:00:00 2001
From: MarcA711 <40744649+MarcA711@users.noreply.github.com>
Date: Tue, 18 Jun 2024 13:20:20 +0000
Subject: [PATCH] fix key error for custom models

---
 frigate/detectors/plugins/rknn.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/frigate/detectors/plugins/rknn.py b/frigate/detectors/plugins/rknn.py
index 49f1f4c02a..af22ca3588 100644
--- a/frigate/detectors/plugins/rknn.py
+++ b/frigate/detectors/plugins/rknn.py
@@ -41,12 +41,12 @@ def __init__(self, config: RknnDetectorConfig):
         if model_props["preset"]:
             config.model.model_type = model_props["model_type"]
 
-        if model_props["model_type"] == ModelTypeEnum.yolonas:
-            logger.info(
-                "You are using yolo-nas with weights from DeciAI. "
-                "These weights are subject to their license and can't be used commercially. "
-                "For more information, see: https://docs.deci.ai/super-gradients/latest/LICENSE.YOLONAS.html"
-            )
+            if model_props["model_type"] == ModelTypeEnum.yolonas:
+                logger.info(
+                    "You are using yolo-nas with weights from DeciAI. "
+                    "These weights are subject to their license and can't be used commercially. "
+                    "For more information, see: https://docs.deci.ai/super-gradients/latest/LICENSE.YOLONAS.html"
+                )
 
         from rknnlite.api import RKNNLite