From 16ec3905274bb63867ccffdf3e23833927ba5a38 Mon Sep 17 00:00:00 2001
From: dongdaxiang <dongdaxiang@baidu.com>
Date: Wed, 29 Apr 2020 23:29:19 +0800
Subject: [PATCH] refine list_model

---
 python/paddle_serving_app/package.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/paddle_serving_app/package.py b/python/paddle_serving_app/package.py
index 42690139..5ed4ef2f 100644
--- a/python/paddle_serving_app/package.py
+++ b/python/paddle_serving_app/package.py
@@ -28,13 +28,13 @@ def parse_args():  # pylint: disable=doc-string-missing
     parser.add_argument(
         "--get_model", type=str, default="", help="Download a specific model")
     parser.add_argument(
-        "--list_model", type=bool, default=False, help="List Models")
+        '--list_model', nargs='*', default=None, help="List Models")
     return parser.parse_args()
 
 
 if __name__ == "__main__":
     args = parse_args()
-    if args.list_model:
+    if args.list_model != None:
         model_handle = ServingModels()
         model_names = model_handle.get_model_list()
         for key in model_names:
-- 
GitLab