diff --git a/aidge_core/unit_tests/test_recipies.py b/aidge_core/unit_tests/test_recipies.py index 754907443530f7e73d1e10ed9549d0c8eb78a011..883cbffc0f22c6a3d009f643dadf0aec9eb3f8fc 100644 --- a/aidge_core/unit_tests/test_recipies.py +++ b/aidge_core/unit_tests/test_recipies.py @@ -34,9 +34,9 @@ class test_recipies(unittest.TestCase): def test_fuse_matmul_add(self): matmul0 = aidge_core.GenericOperator("MatMul", 1, 2, 1, name="MatMul0") - add0 = aidge_core.Add(name="Add0") + add0 = aidge_core.Add(2, name="Add0") matmul1 = aidge_core.GenericOperator("MatMul", 1, 2, 1, name="MatMul1") - add1 = aidge_core.Add(name="Add1") + add1 = aidge_core.Add(2, name="Add1") graph_view = aidge_core.sequential([matmul0, add0, matmul1, add1])