Update predict.py

obss · MLDovakin · Mar 24, 2023 · Mar 24, 2023 · Mar 25, 2023 · Mar 25, 2023
commit 828de94672b559aaba69ac8c13c3a74b68fa0e51
diff --git a/sahi/predict.py b/sahi/predict.py
@@ -310,7 +310,7 @@ def bbox_sort(a, b, thresh):
     return bbox_a[1] - bbox_b[1]
 
 
-def agg_prediction(thresh, result):
+def agg_prediction(thresh, result: PredictionResult):
     coord_list = []
     res = result.object_prediction_list
     for ann in res: