From f66f234517153b7e4b191bce36a189188203a21c Mon Sep 17 00:00:00 2001 From: zhaojichen Date: Thu, 23 Apr 2020 08:26:19 -0400 Subject: [PATCH] change error type --- mindspore/_checkparam.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/_checkparam.py b/mindspore/_checkparam.py index d01de7396a..dc2f71cc18 100644 --- a/mindspore/_checkparam.py +++ b/mindspore/_checkparam.py @@ -315,7 +315,7 @@ class ParamValidator: type_names = [t.__name__ for t in valid_types] num_types = len(valid_types) raise TypeError(f'The type of `{arg_name}` should be {"one of " if num_types > 1 else ""}' - f'{type_names if num_types > 1 else type_names[0]}, but got {type(arg_value).__name__}.') + f'{type_names if num_types > 1 else type_names[0]}, but got {type(arg_value).__name__}.') if isinstance(arg_value, type(mstype.tensor)): arg_value = arg_value.element_type()