diff --git a/aidge_onnx/node_import/onnx_converters/resize.py b/aidge_onnx/node_import/onnx_converters/resize.py index bf77f3ced673cc1ecd7e3122d49ce8819b413555..7e0f7c0b6bf4fe36f6de83f4bc4e5feb621fd996 100644 --- a/aidge_onnx/node_import/onnx_converters/resize.py +++ b/aidge_onnx/node_import/onnx_converters/resize.py @@ -8,7 +8,6 @@ http://www.eclipse.org/legal/epl-2.0. SPDX-License-Identifier: EPL-2.0 """ from typing import List -import numpy as np import aidge_core import onnx @@ -18,8 +17,10 @@ from aidge_onnx.utils import get_node_attributes from aidge_core import Log +from typing import Dict, Tuple + @auto_register_import("resize") -def import_resize(onnx_node: onnx.NodeProto, input_nodes: List[tuple[aidge_core.Node, int]], opset: int = None) -> aidge_core.Node: +def import_resize(onnx_node: onnx.NodeProto, input_nodes: List[Tuple[aidge_core.Node, int]], opset: int = None) -> aidge_core.Node: """ :param onnx_node: ONNX node to convert :type onnx_node: onnx.NodeProto @@ -31,9 +32,9 @@ def import_resize(onnx_node: onnx.NodeProto, input_nodes: List[tuple[aidge_core. node_name = onnx_node.name if onnx_node.name else onnx_node.output[0] onnx_attrs = get_node_attributes(onnx_node, opset) - resize_attrs: dict = {} + resize_attrs: Dict = {} - mode_dict: dict[str, aidge_core.Interpolation.Mode] = { + mode_dict: Dict[str, aidge_core.Interpolation.Mode] = { 'cubic': aidge_core.Interpolation.Mode.CUBIC, 'linear': aidge_core.Interpolation.Mode.LINEAR, 'floor': aidge_core.Interpolation.Mode.FLOOR,