Extract shared inference base types

Move pose-related enums and dataclasses (PoseBackends, WorkerState, PoseSource, PosePacket, PoseResult) out of `dlc_processor.py` into a new `services/inference/base.py` module and import them back into the processor. This centralizes inference contracts for reuse and extends `PosePacket` with optional skeleton metadata fields (`skeleton_id`, `skeleton_edges`).

Author: C-Achard

dlclivegui/services/dlc_processor.py

@@ -10,13 +10,12 @@
 from collections import deque
 from contextlib import contextmanager
 from dataclasses import dataclass
-from enum import Enum, auto
 from typing import Any
 
 import numpy as np
 from PySide6.QtCore import QObject, Signal
 
-from dlclivegui.config import DLC_DO_LOG_TIMING, DLCProcessorSettings, ModelType
+from dlclivegui.config import DLC_DO_LOG_TIMING, DLCProcessorSettings
 from dlclivegui.processors.processor_utils import (
     ProcessorSpec,
     create_spec_from_scan,
@@ -28,6 +27,8 @@
 from dlclivegui.utils.stats import WorkerTimingStats
 from dlclivegui.utils.utils import format_thread_stack
 
+from .inference.base import PoseBackends, PosePacket, PoseResult, PoseSource, WorkerState
+
 logger = logging.getLogger(__name__)
 STOP_WORKER_TIMEOUT = 10.0  # # seconds to wait in STOPPING state before scheduling background reaping
 
@@ -44,41 +45,6 @@
 ENABLE_PROFILING = False
 
 
-class PoseBackends(Enum):
-    DLC_LIVE = auto()
-
-
-class WorkerState(Enum):
-    STOPPED = auto()
-    STARTING = auto()
-    RUNNING = auto()
-    STOPPING = auto()
-    FAULTED = auto()
-
-
-@dataclass
-class PoseResult:
-    pose: np.ndarray | None
-    timestamp: float
-    packet: PosePacket | None = None
-
-
-@dataclass(slots=True, frozen=True)
-class PoseSource:
-    backend: PoseBackends  # e.g. "DLCLive"
-    model_type: ModelType | None = None
-
-
-@dataclass(slots=True, frozen=True)
-class PosePacket:
-    schema_version: int = 0
-    keypoints: np.ndarray | None = None
-    keypoint_names: list[str] | None = None
-    individual_ids: list[str] | None = None
-    source: PoseSource = PoseSource(backend=PoseBackends.DLC_LIVE)
-    raw: Any | None = None
-
-
 def validate_pose_array(
     pose: Any, *, source_backend: PoseBackends | str = PoseBackends.DLC_LIVE, check_finite: bool = True
 ) -> np.ndarray:

dlclivegui/services/inference/__init__.py

@@ -0,0 +1,44 @@
+from dataclasses import dataclass
+from enum import Enum, auto
+from typing import Any
+
+import numpy as np
+
+from dlclivegui.config import ModelType
+
+
+class PoseBackends(Enum):
+    DLC_LIVE = auto()
+
+
+class WorkerState(Enum):
+    STOPPED = auto()
+    STARTING = auto()
+    RUNNING = auto()
+    STOPPING = auto()
+    FAULTED = auto()
+
+
+@dataclass(slots=True, frozen=True)
+class PoseSource:
+    backend: PoseBackends  # e.g. "DLCLive"
+    model_type: ModelType | None = None
+
+
+@dataclass(slots=True, frozen=True)
+class PosePacket:
+    schema_version: int = 0
+    keypoints: np.ndarray | None = None
+    keypoint_names: list[str] | None = None
+    individual_ids: list[str] | None = None
+    skeleton_id: str | None = None
+    skeleton_edges: tuple[tuple[str, str], ...] | None = None
+    source: PoseSource = PoseSource(backend=PoseBackends.DLC_LIVE)
+    raw: Any | None = None
+
+
+@dataclass
+class PoseResult:
+    pose: np.ndarray | None
+    timestamp: float
+    packet: PosePacket | None = None