提交 42aa06ef 编写于 作者: S Steffy-zxf 提交者: wuzewu

Update the command run to inform users some modules that can't do prediction (#15)

上级 fdc6008d
......@@ -98,14 +98,14 @@ class RunCommand(BaseCommand):
module = hub.Module(module_dir=module_dir)
except:
print(
"ERROR! %s is a model. The command is only for the module type but not the model type."
"ERROR! %s is a model. The command run is only for the module type but not the model type."
% module_name)
sys.exit(0)
self.parse_args_with_module(module, argv[1:])
if not module.default_signature:
print("ERROR! Module %s is not callable." % module_name)
print("ERROR! Module %s is not able to predict." % module_name)
return False
if not self.args.signature:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册