Update test_nms_match_small.py

open-mmlab · Oct 8, 2023 · 2fb9984 · 2fb9984
1 parent f203be5
commit 2fb9984
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_ops/test_nms_match_small.py b/tests/test_ops/test_nms_match_small.py
@@ -79,7 +79,8 @@ def forward(self, boxes, scores, iou_threshold, score_threshold):
 
 
 @pytest.mark.skipif(
-    reason='Need to build onnxrumtime custom op', condition=get_ops_path == '')
+    reason='Need to build onnxrumtime custom op',
+    condition=get_ops_path() == '')
 def test_nms_match():
     print('Running compilation...')
     # here is a PyTorch test
@@ -116,3 +117,4 @@ def test_nms_match():
         numpy.array(torch_output),
         numpy.array(ort_output[0])), 'list are not equal'
     os.remove(temp_onnx.name)
+