Skip to content

Commit 2711229

Browse files
author
Matthew Brookhart
authored
[ONNX] enable the onnx tests after PR #8274 merged (#9019)
* enable the onnx tests after PR #8274 merged * fix lint
1 parent fd27aab commit 2711229

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

python/tvm/relay/frontend/onnx.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,11 @@ class onnx_input(list):
6969

7070
def __getitem__(self, item):
7171
if isinstance(item, slice):
72-
indices = list(range(item.stop)[item])
72+
if item.stop is None:
73+
stop = len(self)
74+
else:
75+
stop = item.stop
76+
indices = list(range(stop)[item])
7377
return [self[i] for i in indices]
7478
if isinstance(item, int):
7579
return list(self)[item] if item < len(self) else None

tests/python/frontend/onnx/test_forward.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4145,11 +4145,7 @@ def verify_nms(
41454145
)
41464146

41474147

4148-
# @tvm.testing.parametrize_targets
4149-
@pytest.mark.skip(
4150-
"Test regressed due to not being run in CI"
4151-
+ " tracked here: https://github.com/apache/tvm/pull/8274"
4152-
)
4148+
@tvm.testing.parametrize_targets
41534149
def test_loop(target, dev):
41544150
def verify_cond_loop():
41554151
y_in = helper.make_tensor_value_info("y_in", TensorProto.FLOAT, [1])

0 commit comments

Comments
 (0)