enable the onnx tests after PR apache#8274 merged

mbrookhart · Sep 15, 2021 · 9e06d69 · 9e06d69
1 parent 9bc4dc0
commit 9e06d69
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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])