Skip to content
Snippets Groups Projects
Commit 59887ff3 authored by auphelia's avatar auphelia
Browse files

[Test] Fix bug in setting of idict

parent 3dc8cdf2
No related branches found
No related tags found
No related merge requests found
......@@ -93,7 +93,7 @@ def test_move_flatten_past_affine(data_layout, batch_size):
# compare execution before and after transformation
inp_values = gen_finn_dt_tensor(DataType.INT2, ishape)
idict = {"inp": inp_values}
idict = {model.graph.input[0].name: inp_values}
model_transformed = model.transform(MoveFlattenPastAffine())
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