Skip to content
Snippets Groups Projects
Commit 185022cb authored by auphelia's avatar auphelia
Browse files

[Test] Fix bug in setting of idict

parent 72bc4752
No related branches found
No related tags found
No related merge requests found
......@@ -51,7 +51,7 @@ def test_move_transpose_past_scalar_mul(perm, scalar):
# compare execution before and after transformation
inp_values = np.random.uniform(low=0, high=1, size=(1, 2, 3, 4)).astype(np.float32)
idict = {"inp": inp_values}
idict = {model.graph.input[0].name: inp_values}
model_transformed = model.transform(MoveTransposePastScalarMul())
assert oxe.compare_execution(model, model_transformed, idict)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment