diff --git a/src/maxdiffusion/max_logging.py b/src/maxdiffusion/max_logging.py index 2edb43f4..4a81817c 100644 --- a/src/maxdiffusion/max_logging.py +++ b/src/maxdiffusion/max_logging.py @@ -16,6 +16,10 @@ """Stub for logging utilities. Right now just meant to avoid raw prints""" +import logging + +logger = logging.getLogger(__name__) def log(user_str): + logger.info(user_str) print(user_str, flush=True)