From 56ec492718f284309bbb755a1ad021b965804246 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Fri, 5 Jul 2024 11:31:45 +0200
Subject: [PATCH] Fix softmax binding (wrong argument type)
---
python_binding/operator/pybind_Softmax.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python_binding/operator/pybind_Softmax.cpp b/python_binding/operator/pybind_Softmax.cpp
index f27e469d8..8b6e16d93 100644
--- a/python_binding/operator/pybind_Softmax.cpp
+++ b/python_binding/operator/pybind_Softmax.cpp
@@ -21,7 +21,7 @@ namespace Aidge {
void init_Softmax(py::module& m) {
py::class_<Softmax_Op, std::shared_ptr<Softmax_Op>, OperatorTensor>(m, "SoftmaxOp", py::multiple_inheritance())
- .def(py::init<std::size_t>(), py::arg("axis"))
+ .def(py::init<std::int32_t>(), py::arg("axis"))
.def_static("get_inputs_name", &Softmax_Op::getInputsName)
.def_static("get_outputs_name", &Softmax_Op::getOutputsName);
declare_registrable<Softmax_Op>(m, "SoftmaxOp");
--
GitLab