diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index 48089d164a2f..ec222f7ce867 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 is 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])