Merge pull request #351 from AI-Hypercomputer:ninatu/logging

Google-ML-Automation · Google-ML-Automation · commit 2d237f3f0f9e · 2026-04-14T09:33:19.000-07:00
PiperOrigin-RevId: 899629155
diff --git a/src/maxdiffusion/max_logging.py b/src/maxdiffusion/max_logging.py
@@ -16,6 +16,11 @@
 
 """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)
