Browse Source

fix ut in MindConverter

pull/1271/head
moran 4 years ago
parent
commit
d5bc9674e3
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      tests/ut/mindconverter/graph_based_converter/mapper/test_cumsum.py

+ 3
- 3
tests/ut/mindconverter/graph_based_converter/mapper/test_cumsum.py View File

@@ -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]


Loading…
Cancel
Save