From 9e06d69ce7bb9e3a8440149ffcada41e919e933d Mon Sep 17 00:00:00 2001 From: Matthew Date: Wed, 15 Sep 2021 10:50:38 -0600 Subject: [PATCH 1/2] enable the onnx tests after PR #8274 merged --- python/tvm/relay/frontend/onnx.py | 6 +++++- tests/python/frontend/onnx/test_forward.py | 6 +----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index 48089d164a2f..1634fed5791f 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -69,7 +69,11 @@ class onnx_input(list): def __getitem__(self, item): if isinstance(item, slice): - indices = list(range(item.stop)[item]) + if item.stop == None: + stop = len(self) + else: + stop = item.stop + indices = list(range(stop)[item]) return [self[i] for i in indices] if isinstance(item, int): return list(self)[item] if item < len(self) else None diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py index d9f2e97f8247..d7239b286f48 100644 --- a/tests/python/frontend/onnx/test_forward.py +++ b/tests/python/frontend/onnx/test_forward.py @@ -4145,11 +4145,7 @@ def verify_nms( ) -# @tvm.testing.parametrize_targets -@pytest.mark.skip( - "Test regressed due to not being run in CI" - + " tracked here: https://github.com/apache/tvm/pull/8274" -) +@tvm.testing.parametrize_targets def test_loop(target, dev): def verify_cond_loop(): y_in = helper.make_tensor_value_info("y_in", TensorProto.FLOAT, [1]) From 0e987a0deeeed3715dadffa3ce62f2ffd3c7ebc5 Mon Sep 17 00:00:00 2001 From: Matthew Date: Wed, 15 Sep 2021 11:06:29 -0600 Subject: [PATCH 2/2] fix lint --- python/tvm/relay/frontend/onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index 1634fed5791f..ec222f7ce867 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -69,7 +69,7 @@ class onnx_input(list): def __getitem__(self, item): if isinstance(item, slice): - if item.stop == None: + if item.stop is None: stop = len(self) else: stop = item.stop