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