MAINT: fix type.

oddkiva · Dec 27, 2023 · 5ffdb24 · 5ffdb24
1 parent b20bf95
commit 5ffdb24
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/oddkiva/shakti/inference/coreml/examples/run_yolov4_tiny.py b/python/oddkiva/shakti/inference/coreml/examples/run_yolov4_tiny.py
@@ -83,7 +83,7 @@ def get_yolo_boxes(yolo_out: np.ndarray, yolo_layers: dict['str': Any],
     boxes = [Box(*b) for b in boxes]
     return boxes
 
-def nms(boxes: [Box], iou_thres=0.4):
+def nms(boxes: list[Box], iou_thres=0.4):
     def compare(x: Box, y: Box):
         return y.p_object - x.p_object
     from functools import cmp_to_key