Skip to content

Commit

Permalink
feat: show full tracebacks in debug mode
Browse files Browse the repository at this point in the history
  • Loading branch information
sauyon committed Sep 12, 2022
1 parent 2d03474 commit 59578d2
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion bentoml_cli/utils.py
Expand Up @@ -15,6 +15,7 @@

from bentoml.exceptions import BentoMLException
from bentoml._internal.log import configure_logging
from bentoml._internal.configuration import get_debug_mode
from bentoml._internal.configuration import set_debug_mode
from bentoml._internal.configuration import set_quiet_mode
from bentoml._internal.utils.analytics import track
Expand Down Expand Up @@ -164,7 +165,11 @@ def wrapper(*args: P.args, **kwargs: P.kwargs) -> t.Any:
return func(*args, **kwargs)
except BentoMLException as err:
msg = f"[{cmd_group.name}] `{command_name}` failed: {str(err)}"
raise ClickException(click.style(msg, fg="red")) from err
if get_debug_mode():
ClickException(click.style(msg, fg="red")).show()
raise err from None
else:
raise ClickException(click.style(msg, fg="red")) from err

return t.cast("ClickFunctionWrapper[t.Any]", wrapper)

Expand Down

0 comments on commit 59578d2

Please sign in to comment.