Browse Source

Change the Parameter type name from 'ParameterTensor' to 'Parameter'

r1.7
Zhang Qinghua 4 years ago
parent
commit
973008ebcf
3 changed files with 3 additions and 3 deletions
  1. +1
    -1
      mindspore/python/mindspore/common/parameter.py
  2. +1
    -1
      tests/ut/python/pipeline/parse/test_outermost_net_pass_non_tensor_inputs.py
  3. +1
    -1
      tests/ut/python/pynative_mode/test_outermost_non_tensor_input.py

+ 1
- 1
mindspore/python/mindspore/common/parameter.py View File

@@ -216,7 +216,7 @@ class Parameter(Tensor_):

@staticmethod
def _get_base_class(input_class):
input_class_name = f'Parameter{input_class.__name__}'
input_class_name = Parameter.__name__
if input_class_name in Parameter.__base_type__:
new_type = Parameter.__base_type__[input_class_name]
else:


+ 1
- 1
tests/ut/python/pipeline/parse/test_outermost_net_pass_non_tensor_inputs.py View File

@@ -101,7 +101,7 @@ def test_outermost_net_pass_parameter():
"support bool, int, float, None, tensor, " \
"mstype.Number(mstype.bool, mstype.int, mstype.float, mstype.uint), " \
"and tuple or list containing only these types, and dict whose values are these types, " \
"but the 1th arg type is <class 'mindspore.common.parameter.ParameterTensor'>, " \
"but the 1th arg type is <class 'mindspore.common.parameter.Parameter'>, " \
"value is 'Parameter (name=weight, shape=(2, 2), dtype=Float32, requires_grad=True)'" \
in str(err.value)



+ 1
- 1
tests/ut/python/pynative_mode/test_outermost_non_tensor_input.py View File

@@ -103,7 +103,7 @@ def test_outermost_net_pass_parameter():
"support bool, int, float, None, tensor, " \
"mstype.Number(mstype.bool, mstype.int, mstype.float, mstype.uint), " \
"and tuple or list containing only these types, and dict whose values are these types, " \
"but the 1th arg type is <class 'mindspore.common.parameter.ParameterTensor'>, " \
"but the 1th arg type is <class 'mindspore.common.parameter.Parameter'>, " \
"value is 'Parameter (name=weight, shape=(2, 2), dtype=Float32, requires_grad=True)'" \
in str(err.value)



Loading…
Cancel
Save