Skip to content

Commit b0e048d

Browse files
author
Sergey Shtin
committed
lint fixes.
1 parent 182c64e commit b0e048d

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/python/relay/test_op_level1.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -527,15 +527,17 @@ def test_concatenate3(target, dev):
527527
t_shape = [3, 2, 2, ending]
528528
do_concat_test(shapes, t_shape, dtype, axis, dev, target)
529529

530+
530531
@tvm.testing.parametrize_targets("llvm")
531532
def test_concatenate4(target, dev):
532533
np.random.seed(7)
533534
x_shape = (2, 1)
534-
x = relay.var("x", shape=x_shape, dtype='int64')
535+
x = relay.var("x", shape=x_shape, dtype="int64")
535536
concat = relay.concatenate([x], axis=1)
536537
f = relay.Function([x], concat)
537-
x_val = np.array([[33], [13]], dtype='int64')
538-
op_res = relay.create_executor('graph', device=tvm.cpu(), target='llvm').evaluate(f)(x_val)
538+
x_val = np.array([[33], [13]], dtype="int64")
539+
graph = relay.create_executor("graph", device=tvm.cpu(), target="llvm")
540+
op_res = graph.evaluate(f)(x_val)
539541
ref_res = np.concatenate([x_val], axis=1)
540542
tvm.testing.assert_allclose(op_res.numpy(), ref_res, rtol=0.000001)
541543

0 commit comments

Comments
 (0)