minor fct simplification

Demo.ipynb

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a2a102251661c90c8754034554147835394b71507dc496eb1f09a34665381eb6
-size 979765
+oid sha256:777844f15f935804cd331020a2338c7db3670a96b0dc6d3430b2602cde666e68
+size 989346

post_processing/graphic_elt_pp.py

@@ -89,7 +89,6 @@ def expand_boxes(
 def retrieve_title(
     boxes: npt.NDArray[np.float64],
     labels: npt.NDArray[np.int_],
-    scores: npt.NDArray[np.float64],
     classes: List[str],
 ) -> Tuple[npt.NDArray[np.float64], npt.NDArray[np.int_], npt.NDArray[np.float64]]:
     """
@@ -101,13 +100,10 @@ def retrieve_title(
     Args:
         boxes (numpy.ndarray): Array of bounding boxes with shape (N, 4).
         labels (numpy.ndarray): Array of labels with shape (N,).
-        scores (numpy.ndarray): Array of confidence scores with shape (N,).
         classes (list): List of class labels.
 
     Returns:
-        numpy.ndarray [N x 4]: Array of bounding boxes (unchanged).
-        numpy.ndarray [N]: Array of labels (potentially modified).
-        numpy.ndarray [N]: Array of scores (unchanged).
+        numpy.ndarray [N]: Array of labels.
     """
     if classes.index("chart_title") not in labels:
         widths = boxes[:, 2] - boxes[:, 0]
@@ -115,4 +111,4 @@ def retrieve_title(
         replaced = np.argmax(scores) if max(scores) > 0 else None
         if replaced is not None:
             labels[replaced] = classes.index("chart_title")
-    return boxes, labels, scores
+    return labels