Skip to content

Commit 4bfc93b

Browse files
Print help while there is no arguments for the elasticdl_client command. (#2096)
1 parent 0397c84 commit 4bfc93b

File tree

1 file changed

+20
-4
lines changed

1 file changed

+20
-4
lines changed

elasticdl_client/main.py

Lines changed: 20 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
# limitations under the License.
1313

1414
import argparse
15+
import sys
1516

1617
from elasticdl_client.api import (
1718
build_zoo,
@@ -30,22 +31,33 @@ def build_argument_parser():
3031
subparsers.required = True
3132

3233
# 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+
)
3438
zoo_subparsers = zoo_parser.add_subparsers()
3539
zoo_subparsers.required = True
3640

3741
# 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+
)
3945
zoo_init_parser.set_defaults(func=init_zoo)
4046
args.add_zoo_init_arguments(zoo_init_parser)
4147

4248
# 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+
)
4452
zoo_build_parser.set_defaults(func=build_zoo)
4553
args.add_zoo_build_arguments(zoo_build_parser)
4654

4755
# 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+
)
4961
zoo_push_parser.set_defaults(func=push_zoo)
5062
args.add_zoo_push_arguments(zoo_push_parser)
5163

@@ -78,6 +90,10 @@ def build_argument_parser():
7890

7991
def main():
8092
parser = build_argument_parser()
93+
if len(sys.argv) == 1:
94+
parser.print_help(sys.stderr)
95+
sys.exit(1)
96+
8197
args, _ = parser.parse_known_args()
8298
args.func(args)
8399

0 commit comments

Comments
 (0)