fix(pre_commit): 🎨 auto format pre-commit hooks

Author: pre-commit-ci[bot]

trackers/core/ksp/solver.py

@@ -1,6 +1,6 @@
 from collections import defaultdict
 from dataclasses import dataclass
-from typing import Any, List, Optional, Tuple, Set
+from typing import Any, List, Optional, Set, Tuple
 
 import networkx as nx
 import numpy as np
@@ -39,7 +39,9 @@ def __init__(
         size_weight: float = 0.1,
         conf_weight: float = 0.1,
     ):
-        self.path_overlap_penalty = path_overlap_penalty if path_overlap_penalty is not None else 40  
+        self.path_overlap_penalty = (
+            path_overlap_penalty if path_overlap_penalty is not None else 40
+        )
         self.weight_key = "weight"
         self.source = "SOURCE"
         self.sink = "SINK"
@@ -58,7 +60,9 @@ def __init__(
             self.weights["conf"] = conf_weight
 
         # Entry/exit region settings
-        self.entry_exit_regions: List[Tuple[int, int, int, int]] = []  # (x1, y1, x2, y2)
+        self.entry_exit_regions: List[
+            Tuple[int, int, int, int]
+        ] = []  # (x1, y1, x2, y2)
 
         # Border region settings
         self.use_border_regions = True
@@ -102,7 +106,9 @@ def set_border_entry_exit(
             frame_size (Tuple[int, int]): Size of the image (width, height).
         """
         self.use_border_regions = use_border
-        self.active_borders = borders if borders is not None else {"left", "right", "top", "bottom"}
+        self.active_borders = (
+            borders if borders is not None else {"left", "right", "top", "bottom"}
+        )
         self.border_margin = margin
         self.frame_size = frame_size
 
@@ -139,7 +145,9 @@ def _edge_cost(self, nodeU: TrackNode, nodeV: TrackNode) -> float:
 
         area_a = (bboxU[2] - bboxU[0]) * (bboxU[3] - bboxU[1])
         area_b = (bboxV[2] - bboxV[0]) * (bboxV[3] - bboxV[1])
-        size_penalty = np.log((max(area_a, area_b) / (min(area_a, area_b) + 1e-6)) + 1e-6)
+        size_penalty = np.log(
+            (max(area_a, area_b) / (min(area_a, area_b) + 1e-6)) + 1e-6
+        )
 
         conf_penalty = 1 - min(conf_u, conf_v)
 
@@ -208,7 +216,9 @@ def solve(self, k: Optional[int] = None) -> List[List[TrackNode]]:
                 data[self.weight_key] = base + penalty
 
             try:
-                _, path = nx.single_source_dijkstra(G_mod, self.source, self.sink, weight=self.weight_key)
+                _, path = nx.single_source_dijkstra(
+                    G_mod, self.source, self.sink, weight=self.weight_key
+                )
             except nx.NetworkXNoPath:
                 print(f"No path found from source to sink at {_i}th iteration")
                 break

trackers/core/ksp/tracker.py

@@ -1,6 +1,6 @@
 import os
 from collections import defaultdict
-from typing import Callable, List, Optional, Tuple, Set
+from typing import Callable, List, Optional, Set, Tuple
 
 import cv2
 import numpy as np
@@ -65,9 +65,13 @@ def __init__(
             border_margin (Optional[int]): Border thickness in pixels.
             frame_size (Optional[Tuple[int, int]]): Size of the image (width, height).
         """
-        self.entry_exit_regions = entry_exit_regions if entry_exit_regions is not None else []
+        self.entry_exit_regions = (
+            entry_exit_regions if entry_exit_regions is not None else []
+        )
         self.use_border = use_border
-        self.borders = borders if borders is not None else {"left", "right", "top", "bottom"}
+        self.borders = (
+            borders if borders is not None else {"left", "right", "top", "bottom"}
+        )
         self.border_margin = border_margin
         self.frame_size = frame_size
         self._solver = KSPSolver(
@@ -106,7 +110,7 @@ def _update(self, detections: sv.Detections) -> sv.Detections:
         """
         self._solver.append_frame(detections)
         return detections
-    
+
     def set_entry_exit_regions(self, regions: List[Tuple[int, int, int, int]]) -> None:
         """
         Set rectangular entry/exit zones (x1, y1, x2, y2).
@@ -122,7 +126,9 @@ def set_border_entry_exit(
         frame_size: Optional[Tuple[int, int]] = (1920, 1080),
     ) -> None:
         self.use_border = use_border
-        self.borders = borders if borders is not None else {"left", "right", "top", "bottom"}
+        self.borders = (
+            borders if borders is not None else {"left", "right", "top", "bottom"}
+        )
         self.border_margin = margin
         self.frame_size = frame_size
         self._solver.set_border_entry_exit(
@@ -211,12 +217,12 @@ def track(
         if source.lower().endswith(".mp4"):
             frames_generator = sv.get_video_frames_generator(source_path=source)
             video_info = sv.VideoInfo.from_video_path(video_path=source)
-            
+
             self._solver.set_border_entry_exit(
-                self.use_border, 
-                self.borders, 
-                self.border_margin, 
-                (video_info.width, video_info.height)
+                self.use_border,
+                self.borders,
+                self.border_margin,
+                (video_info.width, video_info.height),
             )
 
             for frame in tqdm(
@@ -248,12 +254,12 @@ def track(
 
                 if not has_set_frame_size:
                     self._solver.set_border_entry_exit(
-                            self.use_border, 
-                            self.borders, 
-                            self.border_margin, 
-                            (width, height)
-                        )
-                    
+                        self.use_border,
+                        self.borders,
+                        self.border_margin,
+                        (width, height),
+                    )
+
                 detections = get_model_detections(image)
                 self._update(detections)
         else: