[Fix] Make Unsqueeze registrable

python_binding/operator/pybind_Unsqueeze.cpp

@@ -23,26 +23,25 @@ void init_Unsqueeze(py::module &m) {
   py::class_<Unsqueeze_Op, std::shared_ptr<Unsqueeze_Op>, OperatorTensor>(
       m, "UnsqueezeOp", py::multiple_inheritance(),
       R"mydelimiter(
-		Initialize an unsqueeze operator.
-		:param axes :   axes to unsqueeze between [-r;r-1] 
-						with r = input_tensor.nbDims() + len(axes)
-		:type axes : :py:class: List[Int]
+            Initialize an unsqueeze operator.
+            :param axes:   axes to unsqueeze between [-r;r-1] with r = input_tensor.nbDims() + len(axes)
+            :type axes: :py:class: List[Int]
 		)mydelimiter")
       // Here we bind the methods of the Unsqueeze_Op that will want to access
       .def("get_inputs_name", &Unsqueeze_Op::getInputsName)
       .def("get_outputs_name", &Unsqueeze_Op::getOutputsName)
-      .def("axes", &Unsqueeze_Op::axes);
-  // Here we bind the constructor of the Unsqueeze Node. We add an argument for
-  // each attribute of the operator (in here we only have 'axes') and the last
-  // argument is the node's name.
+      .def_readonly_static("Type", &Unsqueeze_Op::Type)
+      ;
+
+  declare_registrable<Unsqueeze_Op>(m, "UnsqueezeOp");
+
   m.def("Unsqueeze", &Unsqueeze, py::arg("axes") = std::vector<int8_t>({}),
         py::arg("name") = "",
         R"mydelimiter(
-    Initialize a node containing an unsqueeze operator.
-	:param axes :   axes to unsqueeze between [-r;r-1] 
-					with r = input_tensor.nbDims() + len(axes)
-	:type axes : :py:class: List[Int]
-    :param name : name of the node.
-)mydelimiter");
-}
+            Initialize a node containing an unsqueeze operator.
+            :param axes:   axes to unsqueeze between [-r;r-1] with r = input_tensor.nbDims() + len(axes)
+            :type axes: :py:class: List[Int]
+            :param name: name of the node.
+        )mydelimiter");
+    }
 } // namespace Aidge