From 5a31251aa21293fb9cff8132cdeaea364a99adcd Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Mon, 27 Jul 2020 19:51:47 -0700 Subject: [PATCH] [CI][TEST] Temporary disable nmsv4 test (#6151) --- tests/python/frontend/tensorflow/test_forward.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py index 62829df24d8a..4a4a2cdaeba9 100644 --- a/tests/python/frontend/tensorflow/test_forward.py +++ b/tests/python/frontend/tensorflow/test_forward.py @@ -2051,7 +2051,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)