| @@ -16,7 +16,7 @@ | |||||
| from unittest import TestCase | from unittest import TestCase | ||||
| import numpy as np | import numpy as np | ||||
| from mindinsight.mindconverter.graph_based_converter.mapper.base import ONNXToMindSporeMapper | 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 | 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']['variable_name'] = 'cumsum_op' | ||||
| exchange_msg['var_0']['inputs'] = ['x'] | 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() | code = fragment() | ||||
| init_code = code[0] | init_code = code[0] | ||||