From dde4f719b6d656018bfd161bb6ba09cda2fc1eb3 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Tue, 24 Oct 2023 14:00:56 +0000
Subject: [PATCH] [Operator] Bind setHook and getHook function.
---
python_binding/operator/pybind_Operator.cpp | 2 ++
1 file changed, 2 insertions(+)
diff --git a/python_binding/operator/pybind_Operator.cpp b/python_binding/operator/pybind_Operator.cpp
index 3d18fa3b9..ef02b8aae 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)
;
}
}
--
GitLab