refactor: replace List and Tuple with built-in list and tuple type hints

Author: shoumikchow

bbox_visualizer/core/_utils.py

@@ -2,10 +2,9 @@
 
 import logging
 from contextlib import contextmanager
-from typing import List, Tuple
 
 # Global flag to track warning suppression state
-_warnings_suppressed = False
+_warnings_suppressed: bool = False
 
 def suppress_warnings(suppress: bool = True) -> None:
     """Suppress or enable warning messages from bbox-visualizer.
@@ -36,10 +35,7 @@ def _should_suppress_warning() -> bool:
     """Internal function to check if warnings should be suppressed."""
     return _warnings_suppressed
 
-from typing import List, Tuple
-
-
-def _validate_bbox(bbox: List[int]) -> None:
+def _validate_bbox(bbox: list[int]) -> None:
     """Validate bounding box format and values.
 
     Args:
@@ -59,8 +55,7 @@ def _validate_bbox(bbox: List[int]) -> None:
             "Invalid bounding box coordinates: x_min > x_max or y_min > y_max"
         )
 
-
-def _validate_color(color: Tuple[int, int, int]) -> None:
+def _validate_color(color: tuple[int, int, int]) -> None:
     """Validate BGR color values.
 
     Args:
@@ -72,10 +67,9 @@ def _validate_color(color: Tuple[int, int, int]) -> None:
     if not all(0 <= c <= 255 for c in color):
         raise ValueError("Color values must be between 0 and 255")
 
-
 def _check_and_modify_bbox(
-    bbox: List[int], img_size: Tuple[int, int, int], margin: int = 0
-) -> List[int]:
+    bbox: list[int], img_size: tuple[int, int, int], margin: int = 0
+) -> list[int]:
     """Internal function to check and adjust bounding box coordinates.
 
     .. private::

bbox_visualizer/core/flags.py

@@ -1,7 +1,6 @@
 """Functions for drawing flag and T-shaped labels."""
 
 import logging
-from typing import List, Tuple
 
 import cv2
 import numpy as np
@@ -16,12 +15,12 @@
 def add_T_label(
     img: np.ndarray,
     label: str,
-    bbox: List[int],
+    bbox: list[int],
     size: float = 1,
     thickness: int = 2,
     draw_bg: bool = True,
-    text_bg_color: Tuple[int, int, int] = (255, 255, 255),
-    text_color: Tuple[int, int, int] = (0, 0, 0),
+    text_bg_color: tuple[int, int, int] = (255, 255, 255),
+    text_color: tuple[int, int, int] = (0, 0, 0),
 ) -> np.ndarray:
     """Adds a T-shaped label with a vertical line connecting to the bounding box.
 
@@ -100,13 +99,13 @@ def add_T_label(
 def draw_flag_with_label(
     img: np.ndarray,
     label: str,
-    bbox: List[int],
+    bbox: list[int],
     size: float = 1,
     thickness: int = 2,
     write_label: bool = True,
-    line_color: Tuple[int, int, int] = (255, 255, 255),
-    text_bg_color: Tuple[int, int, int] = (255, 255, 255),
-    text_color: Tuple[int, int, int] = (0, 0, 0),
+    line_color: tuple[int, int, int] = (255, 255, 255),
+    text_bg_color: tuple[int, int, int] = (255, 255, 255),
+    text_color: tuple[int, int, int] = (0, 0, 0),
 ) -> np.ndarray:
     """Draws a flag-like label with a vertical line and text box.
 
@@ -183,11 +182,11 @@ def draw_flag_with_label(
 
 def add_multiple_T_labels(
     img: np.ndarray,
-    labels: List[str],
-    bboxes: List[List[int]],
+    labels: list[str],
+    bboxes: list[list[int]],
     draw_bg: bool = True,
-    text_bg_color: Tuple[int, int, int] = (255, 255, 255),
-    text_color: Tuple[int, int, int] = (0, 0, 0),
+    text_bg_color: tuple[int, int, int] = (255, 255, 255),
+    text_color: tuple[int, int, int] = (0, 0, 0),
 ) -> np.ndarray:
     """Adds multiple T-shaped labels to their corresponding bounding boxes.
 
@@ -226,12 +225,12 @@ def add_multiple_T_labels(
 
 def draw_multiple_flags_with_labels(
     img: np.ndarray,
-    labels: List[str],
-    bboxes: List[List[int]],
+    labels: list[str],
+    bboxes: list[list[int]],
     write_label: bool = True,
-    line_color: Tuple[int, int, int] = (255, 255, 255),
-    text_bg_color: Tuple[int, int, int] = (255, 255, 255),
-    text_color: Tuple[int, int, int] = (0, 0, 0),
+    line_color: tuple[int, int, int] = (255, 255, 255),
+    text_bg_color: tuple[int, int, int] = (255, 255, 255),
+    text_color: tuple[int, int, int] = (0, 0, 0),
 ) -> np.ndarray:
     """Adds multiple flag-like labels to their corresponding bounding boxes.
 

bbox_visualizer/core/labels.py

@@ -1,7 +1,5 @@
 """Functions for adding text labels to bounding boxes."""
 
-from typing import List, Tuple
-
 import cv2
 import numpy as np
 
@@ -13,12 +11,12 @@
 def add_label(
     img: np.ndarray,
     label: str,
-    bbox: List[int],
+    bbox: list[int],
     size: float = 1,
     thickness: int = 2,
     draw_bg: bool = True,
-    text_bg_color: Tuple[int, int, int] = (255, 255, 255),
-    text_color: Tuple[int, int, int] = (0, 0, 0),
+    text_bg_color: tuple[int, int, int] = (255, 255, 255),
+    text_color: tuple[int, int, int] = (0, 0, 0),
     top: bool = True,
 ) -> np.ndarray:
     """Add a label to a bounding box, either above or inside it.
@@ -118,13 +116,13 @@ def add_label(
 
 def add_multiple_labels(
     img: np.ndarray,
-    labels: List[str],
-    bboxes: List[List[int]],
+    labels: list[str],
+    bboxes: list[list[int]],
     size: float = 1,
     thickness: int = 2,
     draw_bg: bool = True,
-    text_bg_color: Tuple[int, int, int] = (255, 255, 255),
-    text_color: Tuple[int, int, int] = (0, 0, 0),
+    text_bg_color: tuple[int, int, int] = (255, 255, 255),
+    text_color: tuple[int, int, int] = (0, 0, 0),
     top: bool = True,
 ) -> np.ndarray:
     """Add multiple labels to their corresponding bounding boxes.

bbox_visualizer/core/rectangle.py

@@ -1,7 +1,5 @@
 """Functions for drawing rectangles on images."""
 
-from typing import List, Tuple
-
 import cv2
 import numpy as np
 
@@ -10,8 +8,8 @@
 
 def draw_rectangle(
     img: np.ndarray,
-    bbox: List[int],
-    bbox_color: Tuple[int, int, int] = (255, 255, 255),
+    bbox: list[int],
+    bbox_color: tuple[int, int, int] = (255, 255, 255),
     thickness: int = 3,
     is_opaque: bool = False,
     alpha: float = 0.5,
@@ -47,8 +45,8 @@ def draw_rectangle(
 
 def draw_multiple_rectangles(
     img: np.ndarray,
-    bboxes: List[List[int]],
-    bbox_color: Tuple[int, int, int] = (255, 255, 255),
+    bboxes: list[list[int]],
+    bbox_color: tuple[int, int, int] = (255, 255, 255),
     thickness: int = 3,
     is_opaque: bool = False,
     alpha: float = 0.5,