diff --git a/unit_tests/scheduler/Test_Scheduler.cpp b/unit_tests/scheduler/Test_Scheduler.cpp index 1fa0e57737553c0e436922dfcc777cdf640ea56b..525dbf43fe9d80550fbd5e089efa4f2cf56cf5f1 100644 --- a/unit_tests/scheduler/Test_Scheduler.cpp +++ b/unit_tests/scheduler/Test_Scheduler.cpp @@ -20,7 +20,7 @@ #include "aidge/scheduler/Scheduler.hpp" #include "aidge/backend/cpu.hpp" -#include "aidge/recipies/GraphViewHelper.hpp" +#include "aidge/recipes/GraphViewHelper.hpp" using namespace Aidge; @@ -372,7 +372,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward compile_gradient(gv); SequentialScheduler scheduler(gv); scheduler.forward(); - auto predictedOutput = gv->getOrderedOutputs()[0]; + auto predictedOutput = gv->getOrderedOutputs()[0].first; std::shared_ptr<Tensor> targetOutput = std::make_shared<Tensor>(Array4D<float, 2, 1, 5, 5>{{{{{0.0f, 1.0f, 1.0f, 2.0f, 2.0f}, @@ -386,4 +386,5 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward {6.0f, 6.0f, 6.0f, 7.0f, 7.0f}, {7.0f, 7.0f, 7.0f, 7.0f, 7.0f}}}}}); - REQUIRE_NOTHROW(scheduler.backward(predictedOutput - targetOutput)); + REQUIRE_NOTHROW(scheduler.backward({targetOutput})); +} \ No newline at end of file