From 2fb9984c51ec891903fc3f423aa2cd18f5d0eca9 Mon Sep 17 00:00:00 2001
From: yinfan98 <32722923+yinfan98@users.noreply.github.com>
Date: Sun, 8 Oct 2023 17:10:28 +0800
Subject: [PATCH] Update test_nms_match_small.py

---
 tests/test_ops/test_nms_match_small.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/test_ops/test_nms_match_small.py b/tests/test_ops/test_nms_match_small.py
index 9dcb46515f..2b96019805 100644
--- 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)
+