diff --git a/netspresso/__init__.py b/netspresso/__init__.py index 33f72544..558504da 100644 --- a/netspresso/__init__.py +++ b/netspresso/__init__.py @@ -1,6 +1,19 @@ +import sys from pathlib import Path -from .netspresso import NPQAI, NetsPresso +from loguru import logger + +# Configure logger format FIRST, before importing other modules +# This ensures the custom format is applied before any module uses logger +logger.remove() +logger.add( + sys.stderr, + format="{time:YYYY-MM-DD HH:mm:ss.SSS} | {level: <8} | - {message}", + level="INFO", + colorize=True, +) + +from .netspresso import NPQAI, NetsPresso # noqa: E402 __all__ = ["NetsPresso", "NPQAI"]