fix: use correct func in test

Smirkey · Dec 27, 2023 · f78c5bc · f78c5bc
1 parent f102e6f
commit f78c5bc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bindings/tests/test_speed.py b/bindings/tests/test_speed.py
@@ -10,6 +10,7 @@
     parallel_giou_distance,
     parallel_iou_distance,
     remove_small_boxes,
+    rtree_nms,
     supported_dtypes,
 )
 
@@ -125,4 +126,4 @@ def test_rtree_nms(benchmark, dtype, generate_boxes):
     boxes = generate_boxes
     boxes = boxes.astype(dtype)
     scores = np.random.random(len(boxes))
-    benchmark(nms, boxes, scores, 0.5, 0.5)
+    benchmark(rtree_nms, boxes, scores, 0.5, 0.5)