|
|
@ -305,9 +305,9 @@ class Validator:
|
|
|
|
msg_prefix = f'For \'{prim_name}\' the' if prim_name else 'The'
|
|
|
|
msg_prefix = f'For \'{prim_name}\' the' if prim_name else 'The'
|
|
|
|
if len(valid_types) == 1:
|
|
|
|
if len(valid_types) == 1:
|
|
|
|
raise TypeError(f'{msg_prefix} type of `{arg_name}` should be {type_names[0]},'
|
|
|
|
raise TypeError(f'{msg_prefix} type of `{arg_name}` should be {type_names[0]},'
|
|
|
|
f' but got {get_typename(arg_type)}.')
|
|
|
|
f' but got {get_typename(arg_type)}.')
|
|
|
|
raise TypeError(f'{msg_prefix} type of `{arg_name}` should be one of {type_names},'
|
|
|
|
raise TypeError(f'{msg_prefix} type of `{arg_name}` should be one of {type_names},'
|
|
|
|
f' but got {get_typename(arg_type)}.')
|
|
|
|
f' but got {get_typename(arg_type)}.')
|
|
|
|
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
@staticmethod
|
|
|
|
def check_float_legal_value(arg_name, arg_value, prim_name):
|
|
|
|
def check_float_legal_value(arg_name, arg_value, prim_name):
|
|
|
@ -418,7 +418,7 @@ class ParamValidator:
|
|
|
|
type_names = [t.__name__ for t in valid_types]
|
|
|
|
type_names = [t.__name__ for t in valid_types]
|
|
|
|
num_types = len(valid_types)
|
|
|
|
num_types = len(valid_types)
|
|
|
|
raise TypeError(f'The type of `{arg_name}` should be {"one of " if num_types > 1 else ""}'
|
|
|
|
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)):
|
|
|
|
if isinstance(arg_value, type(mstype.tensor)):
|
|
|
|
arg_value = arg_value.element_type()
|
|
|
|
arg_value = arg_value.element_type()
|
|
|
|