Skip to content

Commit

Permalink
fix relay nms test
Browse files Browse the repository at this point in the history
  • Loading branch information
masahi committed Apr 5, 2021
1 parent b16f457 commit b04d4dc
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions tests/python/relay/test_op_level5.py
Original file line number Diff line number Diff line change
Expand Up @@ -371,8 +371,6 @@ def verify_nms(
)
if isinstance(z_indices, relay.expr.TupleWrapper):
z_indices = z_indices.astuple()
assert "iou_threshold" in z.astext()
assert "iou_threshold" in z_indices.astext()
zz = run_infer_type(z)
zz_indices = run_infer_type(z_indices)
assert zz.checked_type == relay.ty.TensorType(dshape, "float32")
Expand Down

0 comments on commit b04d4dc

Please sign in to comment.