diff --git a/python_binding/graph/pybind_GraphView.cpp b/python_binding/graph/pybind_GraphView.cpp index 376d61ae85b1166a62a392bee90324cf80d2c0f5..a0b52722aa988b452667e1ee7a364913fd7e01c4 100644 --- a/python_binding/graph/pybind_GraphView.cpp +++ b/python_binding/graph/pybind_GraphView.cpp @@ -30,6 +30,8 @@ void init_GraphView(py::module& m) { :param path: save location :type path: str )mydelimiter") + .def("root_node", &GraphView::rootNode) + .def("set_root_node", &GraphView::setRootNode, py::arg("node")) .def("log_outputs", &GraphView::logOutputs, py::arg("path")) .def("get_ordered_inputs", &GraphView::getOrderedInputs) .def("get_ordered_outputs", &GraphView::getOrderedOutputs)