From 2e1d458ee4536cedb6e1ea3edaa47e62d19a4a93 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Sun, 20 Oct 2024 22:43:11 +0000
Subject: [PATCH] change test_scheduler.py tests back as it was
---
aidge_backend_cpu/unit_tests/test_scheduler.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/aidge_backend_cpu/unit_tests/test_scheduler.py b/aidge_backend_cpu/unit_tests/test_scheduler.py
index b27a54fb..494f3456 100644
--- a/aidge_backend_cpu/unit_tests/test_scheduler.py
+++ b/aidge_backend_cpu/unit_tests/test_scheduler.py
@@ -16,6 +16,7 @@ class test_scheduler(unittest.TestCase):
t = aidge_core.Tensor(np.arange(6, dtype=np.int32) - 3)
+ input_node = aidge_core.Producer(t)
relu = aidge_core.ReLU()
input_node.add_child(relu)
@@ -29,7 +30,7 @@ class test_scheduler(unittest.TestCase):
scheduler = aidge_core.SequentialScheduler(gv)
- scheduler.forward(data=[t])
+ scheduler.forward()
out_tensor = relu.get_operator().get_output(0)
expected_out = [0,0,0,0,1,2]
--
GitLab