From 973008ebcffbea94eedd7c4bc48c6ed590c0a016 Mon Sep 17 00:00:00 2001 From: Zhang Qinghua Date: Tue, 8 Mar 2022 09:27:21 +0800 Subject: [PATCH] Change the Parameter type name from 'ParameterTensor' to 'Parameter' --- mindspore/python/mindspore/common/parameter.py | 2 +- .../pipeline/parse/test_outermost_net_pass_non_tensor_inputs.py | 2 +- .../ut/python/pynative_mode/test_outermost_non_tensor_input.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/mindspore/python/mindspore/common/parameter.py b/mindspore/python/mindspore/common/parameter.py index a4fca804faa..07003dc5e8e 100644 --- a/mindspore/python/mindspore/common/parameter.py +++ b/mindspore/python/mindspore/common/parameter.py @@ -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: diff --git a/tests/ut/python/pipeline/parse/test_outermost_net_pass_non_tensor_inputs.py b/tests/ut/python/pipeline/parse/test_outermost_net_pass_non_tensor_inputs.py index 02bea0e946f..d5b5c303a74 100644 --- a/tests/ut/python/pipeline/parse/test_outermost_net_pass_non_tensor_inputs.py +++ b/tests/ut/python/pipeline/parse/test_outermost_net_pass_non_tensor_inputs.py @@ -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 , " \ + "but the 1th arg type is , " \ "value is 'Parameter (name=weight, shape=(2, 2), dtype=Float32, requires_grad=True)'" \ in str(err.value) diff --git a/tests/ut/python/pynative_mode/test_outermost_non_tensor_input.py b/tests/ut/python/pynative_mode/test_outermost_non_tensor_input.py index dea5b5e46a1..954c5fe704e 100644 --- a/tests/ut/python/pynative_mode/test_outermost_non_tensor_input.py +++ b/tests/ut/python/pynative_mode/test_outermost_non_tensor_input.py @@ -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 , " \ + "but the 1th arg type is , " \ "value is 'Parameter (name=weight, shape=(2, 2), dtype=Float32, requires_grad=True)'" \ in str(err.value)