From fb8acb5b3feae139afd725c85287f36dc6c1fe9c Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Mon, 31 Mar 2025 15:47:01 +0200
Subject: [PATCH] Added missing binding

---
 python_binding/scheduler/pybind_Scheduler.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python_binding/scheduler/pybind_Scheduler.cpp b/python_binding/scheduler/pybind_Scheduler.cpp
index c2fb71103..050af45fd 100644
--- a/python_binding/scheduler/pybind_Scheduler.cpp
+++ b/python_binding/scheduler/pybind_Scheduler.cpp
@@ -59,6 +59,7 @@ void init_Scheduler(py::module& m){
 
     py::class_<SequentialScheduler, std::shared_ptr<SequentialScheduler>, Scheduler>(m, "SequentialScheduler")
     .def(py::init<std::shared_ptr<GraphView>&, bool>(), py::arg("graph_view"), py::arg("reset_cp_model") = true)
+    .def("set_scheduling_policy", &SequentialScheduler::setSchedulingPolicy, py::arg("policy"))
     .def("get_sequential_static_scheduling", &SequentialScheduler::getSequentialStaticScheduling, py::arg("step") = 0)
     .def("generate_memory", &SequentialScheduler::generateMemory, py::arg("inc_producers") = false, py::arg("wrap_around_buffer") = false)
     .def("generate_memory_auto_concat", &SequentialScheduler::generateMemoryAutoConcat, py::arg("inc_producers") = false, py::arg("wrap_around_buffer") = false)
-- 
GitLab