fix: 'type' is not a required value for the schema (#64)

* only print describe if not none * type is not a required field in schema
bentoml · Dec 27, 2021 · 747d5d6 · 747d5d6
1 parent 7b69c8e
commit 747d5d6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/bentoctl/cli/__init__.py b/bentoctl/cli/__init__.py
@@ -141,7 +141,8 @@ def describe(file):
     """
     try:
         info_json = describe_deployment(deployment_config_path=file)
-        pprint(info_json)
+        if info_json is not None:
+            pprint(info_json)
     except BentoctlException as e:
         e.show()
         sys.exit(1)

diff --git a/bentoctl/deployment_config.py b/bentoctl/deployment_config.py
@@ -57,10 +57,10 @@ def remove_help_message(schema):
     for field, rules in schema.items():
         if "help_message" in rules:
             del rules["help_message"]
-        if rules["type"] == "dict":
-            rules["schema"] = remove_help_message(rules["schema"])
-        elif rules["type"] == "list":
-            rules["schema"] = remove_help_message({"list_item": rules["schema"]})[
+        if rules.get("type") == "dict":
+            rules["schema"] = remove_help_message(rules.get("schema"))
+        elif rules.get("type") == "list":
+            rules["schema"] = remove_help_message({"list_item": rules.get("schema")})[
                 "list_item"
             ]
         schema[field] = rules