12
12
# limitations under the License.
13
13
14
14
import argparse
15
+ import sys
15
16
16
17
from elasticdl_client .api import (
17
18
build_zoo ,
@@ -30,22 +31,33 @@ def build_argument_parser():
30
31
subparsers .required = True
31
32
32
33
# Initialize the parser for the `elasticdl zoo` commands
33
- zoo_parser = subparsers .add_parser ("zoo" )
34
+ zoo_parser = subparsers .add_parser (
35
+ "zoo" ,
36
+ help = "Initialize | Build | Push a docker image for the model zoo." ,
37
+ )
34
38
zoo_subparsers = zoo_parser .add_subparsers ()
35
39
zoo_subparsers .required = True
36
40
37
41
# elasticdl zoo init
38
- zoo_init_parser = zoo_subparsers .add_parser ("init" )
42
+ zoo_init_parser = zoo_subparsers .add_parser (
43
+ "init" , help = "Initialize the model zoo."
44
+ )
39
45
zoo_init_parser .set_defaults (func = init_zoo )
40
46
args .add_zoo_init_arguments (zoo_init_parser )
41
47
42
48
# elasticdl zoo build
43
- zoo_build_parser = zoo_subparsers .add_parser ("build" )
49
+ zoo_build_parser = zoo_subparsers .add_parser (
50
+ "build" , help = "Build a docker image for the model zoo."
51
+ )
44
52
zoo_build_parser .set_defaults (func = build_zoo )
45
53
args .add_zoo_build_arguments (zoo_build_parser )
46
54
47
55
# elasticdl zoo push
48
- zoo_push_parser = zoo_subparsers .add_parser ("push" )
56
+ zoo_push_parser = zoo_subparsers .add_parser (
57
+ "push" ,
58
+ help = "Push the docker image to a remote registry for the distributed"
59
+ "ElasticDL job." ,
60
+ )
49
61
zoo_push_parser .set_defaults (func = push_zoo )
50
62
args .add_zoo_push_arguments (zoo_push_parser )
51
63
@@ -78,6 +90,10 @@ def build_argument_parser():
78
90
79
91
def main ():
80
92
parser = build_argument_parser ()
93
+ if len (sys .argv ) == 1 :
94
+ parser .print_help (sys .stderr )
95
+ sys .exit (1 )
96
+
81
97
args , _ = parser .parse_known_args ()
82
98
args .func (args )
83
99
0 commit comments