[TVMC] fail gracefully in case no subcommand is provided (apache#6625)

jinqiua · Oct 6, 2020 · 82289ea · 82289ea
1 parent dec5030
commit 82289ea
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/tvm/driver/tvmc/main.py b/python/tvm/driver/tvmc/main.py
@@ -79,7 +79,10 @@ def _main(argv):
         sys.stdout.write("%s\n" % version)
         return 0
 
-    assert hasattr(args, "func"), "Error: missing 'func' attribute for subcommand {0}".format(argv)
+    if not hasattr(args, "func"):
+        # In case no valid subcommand is provided, show usage and exit
+        parser.print_help(sys.stderr)
+        return 1
 
     try:
         return args.func(args)