diff --git a/python_binding/operator/pybind_Operator.cpp b/python_binding/operator/pybind_Operator.cpp index 3d18fa3b96087ae78c14d02bfd1b7959e3a6f693..ef02b8aaef9f4ea3bd97559ad9e94c38c5b1d29e 100644 --- a/python_binding/operator/pybind_Operator.cpp +++ b/python_binding/operator/pybind_Operator.cpp @@ -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) ; } }