Only import degirum module if using degirum detector (#19802)

This commit is contained in:
Josh Hawkins
2025-08-27 08:31:01 -06:00
committed by GitHub
parent 0febc4d456
commit d78b6e528b
+5 -1
View File
@@ -1,7 +1,6 @@
import logging
import queue
import degirum as dg
import numpy as np
from pydantic import Field
from typing_extensions import Literal
@@ -26,6 +25,11 @@ class DGDetector(DetectionApi):
type_key = DETECTOR_KEY
def __init__(self, detector_config: DGDetectorConfig):
try:
import degirum as dg
except ModuleNotFoundError:
raise ImportError("Unable to import DeGirum detector.")
self._queue = queue.Queue()
self._zoo = dg.connect(
detector_config.location, detector_config.zoo, detector_config.token