Remove mlperf logging

coolkp · coolkp · commit 4686b2e51c3b · 2025-07-27T16:25:29.000Z
Signed-off-by: Kunjan patel &lt;kunjanp@google.com&gt;
diff --git a/src/maxdiffusion/train_flux.py b/src/maxdiffusion/train_flux.py
@@ -18,7 +18,10 @@
 
 import jax
 from absl import app
-from maxdiffusion import (max_logging, pyconfig)
+from maxdiffusion import (
+    max_logging,
+    pyconfig,
+)
 
 from maxdiffusion.train_utils import (
     validate_train_config,
