From 25e11cc2e736ad9242762bc97256a95f77e203df Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Tue, 17 Sep 2024 12:38:29 +0000
Subject: [PATCH] Make MetaOperator registrable in Python.
---
python_binding/operator/pybind_MetaOperatorDefs.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/python_binding/operator/pybind_MetaOperatorDefs.cpp b/python_binding/operator/pybind_MetaOperatorDefs.cpp
index 5f6cc12fc..aa1889c69 100644
--- a/python_binding/operator/pybind_MetaOperatorDefs.cpp
+++ b/python_binding/operator/pybind_MetaOperatorDefs.cpp
@@ -194,8 +194,8 @@ void init_MetaOperatorDefs(py::module &m) {
// declare_PaddedMaxPoolingOp<3>(m);
declare_LSTMOp(m);
- py::class_<MetaOperator_Op, std::shared_ptr<MetaOperator_Op>, OperatorTensor>(m, "MetaOperator_Op", py::multiple_inheritance())
- .def(py::init<const char *, const std::shared_ptr<GraphView>&, const std::vector<InputCategory>&>(),
+ py::class_<MetaOperator_Op, std::shared_ptr<MetaOperator_Op>, OperatorTensor>(m, "MetaOperatorOp", py::multiple_inheritance())
+ .def(py::init<const char *, const std::shared_ptr<GraphView>&>(),
py::arg("type"),
py::arg("graph"),
py::arg("forced_inputs_category") = std::vector<InputCategory>())
--
GitLab