Skip to content

Commit 561dcd1

Browse files
committed
Added some python bindings for Node Inputs (getParent and possibleDatatypes)
1 parent b03248d commit 561dcd1

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/pipeline/node/NodeBindings.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,9 @@ void NodeBindings::bind(pybind11::module& m, void* pCallstack){
211211
.def_readwrite("group", &Node::Input::group, DOC(dai, Node, Input, group))
212212
.def_readwrite("name", &Node::Input::name, DOC(dai, Node, Input, name))
213213
.def_readwrite("type", &Node::Input::type, DOC(dai, Node, Input, type))
214+
.def_readwrite("possibleDatatypes", &Node::Input::possibleDatatypes, DOC(dai, Node, Input, possibleDatatypes))
215+
.def("getParent", static_cast<const Node& (Node::Input::*)() const>(&Node::Input::getParent), py::return_value_policy::reference_internal, DOC(dai, Node, Input, getParent))
216+
.def("getParent", static_cast<Node& (Node::Input::*)()>(&Node::Input::getParent), py::return_value_policy::reference_internal, DOC(dai, Node, Input, getParent))
214217
.def("setBlocking", &Node::Input::setBlocking, py::arg("blocking"), DOC(dai, Node, Input, setBlocking))
215218
.def("getBlocking", &Node::Input::getBlocking, DOC(dai, Node, Input, getBlocking))
216219
.def("setQueueSize", &Node::Input::setQueueSize, py::arg("size"), DOC(dai, Node, Input, setQueueSize))

0 commit comments

Comments
 (0)