From 445eb6372f8ebd64221d977ac45728028b9dc62c Mon Sep 17 00:00:00 2001 From: NAUD Maxence <maxence.naud@cea.fr> Date: Sun, 20 Oct 2024 13:44:27 +0000 Subject: [PATCH] [TMP] change scheduler test for Merge to be possible --- aidge_backend_cpu/unit_tests/test_scheduler.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/aidge_backend_cpu/unit_tests/test_scheduler.py b/aidge_backend_cpu/unit_tests/test_scheduler.py index 6ba5a52a..b27a54fb 100644 --- a/aidge_backend_cpu/unit_tests/test_scheduler.py +++ b/aidge_backend_cpu/unit_tests/test_scheduler.py @@ -17,9 +17,12 @@ class test_scheduler(unittest.TestCase): t = aidge_core.Tensor(np.arange(6, dtype=np.int32) - 3) relu = aidge_core.ReLU() + input_node.add_child(relu) gv = aidge_core.GraphView() gv.add(relu) + gv.add(input_node) + gv.set_datatype(aidge_core.dtype.int32) gv.set_backend("cpu") -- GitLab