From 3c4f0b2ba36c2632544c5a06c867987c2ecfcba2 Mon Sep 17 00:00:00 2001 From: Tibo De Peuter Date: Wed, 1 Apr 2026 19:34:47 +0200 Subject: [PATCH] fix(train): disable tqdm in batch mode to prevent log spam --- src/train.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/train.py b/src/train.py index ac6849e..c1a646d 100644 --- a/src/train.py +++ b/src/train.py @@ -1,4 +1,5 @@ import random +import sys import time from dataclasses import asdict from functools import partial @@ -244,7 +245,10 @@ def train(args: PPOArgs): ) print("Starting training...") - iters_bar = tqdm.tqdm(range(1, args.num_iterations + 1)) + iters_bar = tqdm.tqdm( + range(1, args.num_iterations + 1), + disable=not sys.stdout.isatty(), + ) losses = [] for _ in iters_bar: iteration_time_start = time.time()