From c0c587e19635f1e91e476e119ecf916a444e6ff9 Mon Sep 17 00:00:00 2001 From: Olivier BICHLER <olivier.bichler@cea.fr> Date: Thu, 20 Jun 2024 20:01:28 +0200 Subject: [PATCH] Fixed missing input in unit test --- aidge_core/unit_tests/test_impl.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aidge_core/unit_tests/test_impl.py b/aidge_core/unit_tests/test_impl.py index 6e0c1f9b9..26d60f2fb 100644 --- a/aidge_core/unit_tests/test_impl.py +++ b/aidge_core/unit_tests/test_impl.py @@ -41,6 +41,7 @@ class test_OperatorImpl(unittest.TestCase): generic_matmul_op = matmul.get_operator() generic_matmul_op.set_forward_dims(lambda x: x) generic_matmul_op.set_impl(testImpl(generic_matmul_op)) + generic_matmul_op.set_input(0, aidge_core.Tensor(np.arange(18).reshape(1,2,3,3))) generic_matmul_op.forward() self.assertEqual(GLOBAL_CPT, 1) -- GitLab