diff --git a/aidge_quantization/freezeProducers.py b/aidge_quantization/freezeProducers.py index a6ee48437981b9083bebf78cd4dee629a7875539..87839718cce84fa233504aca712c30595797a307 100644 --- a/aidge_quantization/freezeProducers.py +++ b/aidge_quantization/freezeProducers.py @@ -34,5 +34,5 @@ def freeze_weights(graphview: aidge_core.GraphView, all_producers: bool = False) while(parent_node.type() != "Producer"): parent_node = parent_node.get_parent(0) if parent_node is None: - raise ValueError ("error, producer not found") + raise RuntimeError(f"Could not find a parent producer for node {node.name()}") parent_node.get_operator().attr.set_attr("constant",True) diff --git a/python_binding/recipes/pybind_ONNXRecipes.cpp b/python_binding/recipes/pybind_ONNXRecipes.cpp index d1ea339291dea98d042257bf33792f9507606430..9664f76782dcdbfff8735e59cdb178b0753bc993 100644 --- a/python_binding/recipes/pybind_ONNXRecipes.cpp +++ b/python_binding/recipes/pybind_ONNXRecipes.cpp @@ -21,7 +21,7 @@ namespace Aidge { void init_ONNXRecipes(py::module &m) { - m.def("quantize_matching_to_export", &quantizeMatchingtoExport, py::arg("graphView"), py::arg("qop")=true,py::arg("fold_weights")=true); + m.def("quantize_matching_to_export", &quantizeMatchingtoExport, py::arg("graph_view"), py::arg("qop")=true, py::arg("fold_weights")=true); }