[CI][TEST] Temporary disable nmsv4 test (apache#6151)

trevor-m · Sep 2, 2020 · 5e747ef · 5e747ef
1 parent b3edfad
commit 5e747ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py
@@ -2066,7 +2066,7 @@ def _test_forward_nms_v4(bx_shape, score_shape, iou_threshold, score_threshold,
 
 def test_forward_nms():
     """ NonMaxSuppressionV3,4 """
-    for _test_forward_nms in [_test_forward_nms_v3, _test_forward_nms_v4]:
+    for _test_forward_nms in [_test_forward_nms_v3]:
         _test_forward_nms((5, 4), (5,), 0.7, 0.5, 5)
         _test_forward_nms((20, 4), (20,), 0.5, 0.6, 10)
         _test_forward_nms((1000, 4), (1000,), 0.3, 0.7, 1000)