From 59887ff3159ea8dacb21bc8e8449f5d819f0fd61 Mon Sep 17 00:00:00 2001
From: auphelia <jakobapk@web.de>
Date: Thu, 25 Jun 2020 15:49:50 +0100
Subject: [PATCH] [Test] Fix bug in setting of idict
---
tests/transformation/test_move_flatten_past_affine.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/transformation/test_move_flatten_past_affine.py b/tests/transformation/test_move_flatten_past_affine.py
index e5588965c..b2d5e5161 100644
--- a/tests/transformation/test_move_flatten_past_affine.py
+++ b/tests/transformation/test_move_flatten_past_affine.py
@@ -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)
--
GitLab