fix: show warning instead of hard failure when model is missing (#4693)

Signed-off-by: Frost Ming <me@frostming.com>
bentoml · May 6, 2024 · 6de2041 · 6de2041
1 parent e2c21b8
commit 6de2041
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/src/bentoml/_internal/bento/bento.py b/src/bentoml/_internal/bento/bento.py
@@ -420,7 +420,15 @@ def total_size(
                     continue
                 except NotFound:
                     pass
-            global_model = model_store.get(model.tag)
+            try:
+                global_model = model_store.get(model.tag)
+            except NotFound:
+                logger.warning(
+                    "Bento: %s: Missing model %s",
+                    self.tag,
+                    model.tag,
+                )
+                continue
             total_size += global_model.file_size
         return total_size
 

diff --git a/src/bentoml/_internal/log.py b/src/bentoml/_internal/log.py
@@ -23,15 +23,22 @@ def filter(self, record: logging.LogRecord) -> bool:
     "version": 1,
     "disable_existing_loggers": True,
     "filters": {"infofilter": {"()": InfoFilter}},
+    "formatters": {
+        "simple": {
+            "format": "%(levelname)s: %(message)s",
+        }
+    },
     "handlers": {
         "bentomlhandler": {
             "class": "logging.StreamHandler",
             "filters": ["infofilter"],
             "stream": "ext://sys.stdout",
+            "formatter": "simple",
         },
         "defaulthandler": {
             "class": "logging.StreamHandler",
             "level": logging.WARNING,
+            "formatter": "simple",
         },
     },
     "loggers": {