Fix for pylint Failed

test=develop
revert-15774-anakin_subgraph_engine
Krzysztof Binias 6 years ago
parent 1578c60bdd
commit 309ea6f2de

@ -121,9 +121,7 @@ class TestMKLDNNReluPrimitivesAlreadyExist(unittest.TestCase):
block = program.global_block() block = program.global_block()
for name in ground_truth: for name in ground_truth:
block.create_var( block.create_var(
name=name, name=name, dtype='float32', shape=ground_truth[name].shape)
dtype='float32',
shape=ground_truth[name].shape)
relu_op = block.append_op( relu_op = block.append_op(
type="abs", type="abs",
@ -149,8 +147,10 @@ class TestMKLDNNReluPrimitivesAlreadyExist(unittest.TestCase):
# Do at least 2 iterations # Do at least 2 iterations
for i in range(2): for i in range(2):
out = exe.run(program, out = exe.run(
feed={name: var_dict[name] for name in ['x', 'out@GRAD']}, program,
feed={name: var_dict[name]
for name in ['x', 'out@GRAD']},
fetch_list=['x@GRAD']) fetch_list=['x@GRAD'])
self.__assert_close(x_grad, out[0], "x@GRAD") self.__assert_close(x_grad, out[0], "x@GRAD")

Loading…
Cancel
Save