Minor changes

include/aidge/scheduler/Scheduler.hpp

@@ -135,16 +135,6 @@ public:
      */
     void tagConditionalNodes() const;
 
-    /**
-     * @brief Check if the conditional node is required (if one of its conditions
-     * is true).
-     * 
-     * @param node Node to check the condition.
-     * @return true If any node condition is true, meaning it has to be executed.
-     * @return false If all node conditions are false, meaning it can be skipped.
-     */
-    bool isConditionalNodeRequired(NodePtr node) const;
-
     /**
      * @brief Get the static scheduling order of nodes.
      * @param step The step of the static schedule to retrieve (default is 0).
@@ -220,6 +210,16 @@ public:
 
 
 protected:
+    /**
+     * @brief Check if the conditional node is required (if one of its conditions
+     * is true).
+     * 
+     * @param node Node to check the condition.
+     * @return true If any node condition is true, meaning it has to be executed.
+     * @return false If all node conditions are false, meaning it can be skipped.
+     */
+    bool isConditionalNodeRequired(NodePtr node) const;
+
     /**
      * @brief Getter for the set of children Nodes of the given input Nodes.
      * @param producers Set of Nodes for which we want to obtain the set of children Nodes.

python_binding/scheduler/pybind_Scheduler.cpp

@@ -30,6 +30,7 @@ void init_Scheduler(py::module& m){
     py::class_<Scheduler, std::shared_ptr<Scheduler>>(m, "Scheduler")
     .def(py::init<std::shared_ptr<GraphView>&>(), py::arg("graph_view"))
     .def("graph_view", &Scheduler::graphView)
+    .def("tag_conditional_nodes", &Scheduler::tagConditionalNodes)
     .def("save_scheduling_diagram", &Scheduler::saveSchedulingDiagram, py::arg("file_name"))
     .def("save_static_scheduling_diagram", &Scheduler::saveStaticSchedulingDiagram, py::arg("file_name"))
     .def("save_factorized_static_scheduling_diagram", &Scheduler::saveFactorizedStaticSchedulingDiagram, py::arg("file_name"), py::arg("min_repeat") = 2)