Catch hailo initialization error (#12558)

blakeblackshear · Jul 26, 2024 · 1c617ed · 1c617ed
1 parent 979f4ac
commit 1c617ed
Showing 1 changed file with 16 additions and 11 deletions.
diff --git a/frigate/detectors/plugins/hailo8l.py b/frigate/detectors/plugins/hailo8l.py
@@ -3,17 +3,22 @@
 import urllib.request
 
 import numpy as np
-from hailo_platform import (
-    HEF,
-    ConfigureParams,
-    FormatType,
-    HailoRTException,
-    HailoStreamInterface,
-    InferVStreams,
-    InputVStreamParams,
-    OutputVStreamParams,
-    VDevice,
-)
+
+try:
+    from hailo_platform import (
+        HEF,
+        ConfigureParams,
+        FormatType,
+        HailoRTException,
+        HailoStreamInterface,
+        InferVStreams,
+        InputVStreamParams,
+        OutputVStreamParams,
+        VDevice,
+    )
+except ModuleNotFoundError:
+    pass
+
 from pydantic import BaseModel, Field
 from typing_extensions import Literal