Cleanup

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(

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
 

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"