From d5bc9674e372b156ae1d0ffd45ee7e5653128b58 Mon Sep 17 00:00:00 2001 From: moran Date: Mon, 29 Mar 2021 15:40:44 +0800 Subject: [PATCH] fix ut in MindConverter --- .../graph_based_converter/mapper/test_cumsum.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/ut/mindconverter/graph_based_converter/mapper/test_cumsum.py b/tests/ut/mindconverter/graph_based_converter/mapper/test_cumsum.py index 1026d51e..b25b66c4 100644 --- a/tests/ut/mindconverter/graph_based_converter/mapper/test_cumsum.py +++ b/tests/ut/mindconverter/graph_based_converter/mapper/test_cumsum.py @@ -16,7 +16,7 @@ from unittest import TestCase import numpy as np from mindinsight.mindconverter.graph_based_converter.mapper.base import ONNXToMindSporeMapper -from mindinsight.mindconverter.graph_based_converter.common.code_fragment import NewFragment +from mindinsight.mindconverter.graph_based_converter.common.code_fragment import Fragment from mindinsight.mindconverter.graph_based_converter.third_party_graph.onnx_utils import NodeWeight @@ -38,8 +38,8 @@ class TestCumSum(TestCase): exchange_msg['var_0']['variable_name'] = 'cumsum_op' exchange_msg['var_0']['inputs'] = ['x'] - fragment = NewFragment(data_entity=exchange_msg, code_template=template, outputs=outputs_lists, - outputs_mapping=outputs_mapping) + fragment = Fragment(data_entity=exchange_msg, code_template=template, outputs=outputs_lists, + outputs_mapping=outputs_mapping) code = fragment() init_code = code[0]