diff --git a/src/bentoml/_internal/bento/bento.py b/src/bentoml/_internal/bento/bento.py index 623b344dbd..41e15702b9 100644 --- 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 index 57b1545b4d..68dbb26221 100644 --- 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": {