diff --git a/frigate/config/classification.py b/frigate/config/classification.py index c4aaabd2d..572c70e23 100644 --- a/frigate/config/classification.py +++ b/frigate/config/classification.py @@ -3,8 +3,6 @@ from typing import Dict, List, Optional from pydantic import ConfigDict, Field -from frigate.types import ObjectClassificationType - from .base import FrigateBaseModel __all__ = [ @@ -36,6 +34,11 @@ class TriggerAction(str, Enum): NOTIFICATION = "notification" +class ObjectClassificationType(str, Enum): + sub_label = "sub_label" + attribute = "attribute" + + class AudioTranscriptionConfig(FrigateBaseModel): enabled: bool = Field(default=False, title="Enable audio transcription.") language: str = Field( diff --git a/frigate/data_processing/real_time/custom_classification.py b/frigate/data_processing/real_time/custom_classification.py index 434e8b500..71eb1cd87 100644 --- a/frigate/data_processing/real_time/custom_classification.py +++ b/frigate/data_processing/real_time/custom_classification.py @@ -15,10 +15,12 @@ from frigate.comms.event_metadata_updater import ( ) from frigate.comms.inter_process import InterProcessRequestor from frigate.config import FrigateConfig -from frigate.config.classification import CustomClassificationConfig +from frigate.config.classification import ( + CustomClassificationConfig, + ObjectClassificationType, +) from frigate.const import CLIPS_DIR, MODEL_CACHE_DIR from frigate.log import redirect_output_to_logger -from frigate.types import ObjectClassificationType from frigate.util.builtin import EventsPerSecond, InferenceSpeed, load_labels from frigate.util.object import box_overlaps, calculate_region diff --git a/frigate/types.py b/frigate/types.py index db5d927bb..a9e27ba90 100644 --- a/frigate/types.py +++ b/frigate/types.py @@ -25,11 +25,6 @@ class ModelStatusTypesEnum(str, Enum): complete = "complete" -class ObjectClassificationType(str, Enum): - sub_label = "sub_label" - attribute = "attribute" - - class TrackedObjectUpdateTypesEnum(str, Enum): description = "description" face = "face"