From 3fbd66cdec15b90d841833026550f87f4ab9a96d Mon Sep 17 00:00:00 2001 From: Arjun Krishnakumar Date: Mon, 29 Jun 2026 15:16:37 +0200 Subject: [PATCH] fix: log INFO only on the main rank --- src/post_training/utils/logging.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/post_training/utils/logging.py b/src/post_training/utils/logging.py index 4157746..0abcfb3 100644 --- a/src/post_training/utils/logging.py +++ b/src/post_training/utils/logging.py @@ -3,6 +3,7 @@ from __future__ import annotations import logging +import os import sys @@ -11,6 +12,12 @@ def setup_logging(level: int = logging.INFO) -> None: Call once at the beginning of every entry-point script. """ + rank = int(os.environ.get("RANK", 0)) + if level == logging.INFO and rank > 0: + # Suppress INFO logs from non-main ranks to reduce log spam. + logging.basicConfig(handlers=[logging.NullHandler()], force=True) + return + logging.basicConfig( level=level, format="%(asctime)s | %(levelname)-8s | %(name)s | %(message)s",