Skip to content
Snippets Groups Projects
Commit dde4f719 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

[Operator] Bind setHook and getHook function.

parent adf85efb
No related branches found
No related tags found
1 merge request!40PTQ updates
Pipeline #33328 passed
......@@ -29,6 +29,8 @@ void init_Operator(py::module& m){
.def("forward", &Operator::forward)
// py::keep_alive forbide Python to garbage collect implementation will the Operator is not garbade collected !
.def("set_impl", &Operator::setImpl, py::arg("implementation"), py::keep_alive<1, 2>())
.def("get_hook", &Operator::getHook)
.def("add_hook", &Operator::addHook)
;
}
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment