From 46767f41bea7c307ee4219e2a030fe6b92226c24 Mon Sep 17 00:00:00 2001
From: Vincent TEMPLIER <vincent.templier@cea.fr>
Date: Thu, 21 Sep 2023 08:33:13 +0000
Subject: [PATCH] Add getChildren and getParents binding methods for Node
---
python_binding/graph/pybind_Node.cpp | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/python_binding/graph/pybind_Node.cpp b/python_binding/graph/pybind_Node.cpp
index 62b869820..e3666d247 100644
--- a/python_binding/graph/pybind_Node.cpp
+++ b/python_binding/graph/pybind_Node.cpp
@@ -136,6 +136,16 @@ void init_Node(py::module& m) {
:rtype: int
)mydelimiter")
+ .def("get_parents", &Node::getParents,
+ R"mydelimiter(
+ Get parents.
+ )mydelimiter")
+
+ .def("get_children", (std::set<std::shared_ptr<Node>> (Node::*)() const) &Node::getChildren,
+ R"mydelimiter(
+ Get children.
+ )mydelimiter")
+
.def("__call__", &Node::operator(), py::arg("connectors"));
}
} // namespace Aidge
--
GitLab