diff --git a/aidge_onnx/onnx_import.py b/aidge_onnx/onnx_import.py index 9e920ccd343b5d425740a7b37c6d449e40c9e206..a8de870d0df2d0ad82a4113f54fdafa141061cf3 100644 --- a/aidge_onnx/onnx_import.py +++ b/aidge_onnx/onnx_import.py @@ -15,7 +15,7 @@ import colorama from onnx import numpy_helper import onnx from .node_import import ONNX_NODE_CONVERTER_, generic -from .utils import onnx_to_aidge_convert_model_names +from .utils import onnx_to_aidge_model_names def load_onnx(filename: str, verbose: bool = False): """Load an ONNX file and convert it into a :py:class:`aidge_core.GraphView`. @@ -107,7 +107,7 @@ def _load_onnx2graphview(model:onnx.ModelProto, verbose:bool = False): # Clean model if some issues in the model # might affect Aidge in the next steps - model = onnx_to_aidge_convert_model_names(model) + model = onnx_to_aidge_model_names(model) if verbose : print(f"\nGetting Initializers\n====================") # Get the initializers for i in model.graph.initializer: