diff --git a/src/bub/__main__.py b/src/bub/__main__.py index 430baec5..9d50eaa1 100644 --- a/src/bub/__main__.py +++ b/src/bub/__main__.py @@ -2,22 +2,26 @@ from __future__ import annotations +import sys + import typer from bub.framework import BubFramework def _instrument_bub() -> None: + from loguru import logger + + logger.remove() + logger.add(sys.stderr, colorize=True) + try: import logfire logfire.configure() + logger.add(logfire.loguru_handler()) except ImportError: pass - else: - from loguru import logger - - logger.configure(handlers=[logfire.loguru_handler()]) def create_cli_app() -> typer.Typer: