From 955b483cf53b416af8afd3f9489b8c6fc82f5613 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 17 Apr 2025 11:18:44 -0600 Subject: [PATCH] Improve typing --- frigate/detectors/detection_api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/frigate/detectors/detection_api.py b/frigate/detectors/detection_api.py index 403bce574b..ccccc1b65d 100644 --- a/frigate/detectors/detection_api.py +++ b/frigate/detectors/detection_api.py @@ -4,7 +4,7 @@ from typing import List import numpy as np -from frigate.detectors.detector_config import ModelTypeEnum +from frigate.detectors.detector_config import BaseDetectorConfig, ModelTypeEnum logger = logging.getLogger(__name__) @@ -14,7 +14,7 @@ class DetectionApi(ABC): supported_models: List[ModelTypeEnum] @abstractmethod - def __init__(self, detector_config): + def __init__(self, detector_config: BaseDetectorConfig): self.detector_config = detector_config self.thresh = 0.5 self.height = detector_config.model.height