def validate_class_args(self, class_args, algo_type, builtin_name):
if not builtin_name or not class_args:
return
meta = get_builtin_algo_meta(algo_type+'s', builtin_name)
if meta and 'accept_class_args' in meta and meta['accept_class_args'] == False:
raise SchemaError('classArgs is not allowed.')
validator = create_validator_instance(algo_type+'s', builtin_name)
if validator:
try:
validator.validate_class_args(**class_args)
except Exception as e:
raise SchemaError(str(e))
def validate_class_args(self, class_args, algo_type, builtin_name):
if not builtin_name or not class_args:
return
meta = get_builtin_algo_meta(algo_type+'s', builtin_name)
if meta and 'accept_class_args' in meta and meta['accept_class_args'] == False:
raise SchemaError('classArgs is not allowed.')
logging.getLogger('nni.protocol').setLevel(logging.ERROR) # we know IPC is not there, don't complain
validator = create_validator_instance(algo_type+'s', builtin_name)
if validator:
try:
validator.validate_class_args(**class_args)
except Exception as e:
raise SchemaError(str(e))