From d1939f894dd174d44fe2a29e58683fdcaeaec977 Mon Sep 17 00:00:00 2001
From: Lin Chai <linchai@google.com>
Date: Tue, 3 Mar 2026 20:20:12 -0800
Subject: [PATCH] [Tunix] Reduce log spam from type mismatch warnings.

PiperOrigin-RevId: 878247934
---
 tunix/generate/utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tunix/generate/utils.py b/tunix/generate/utils.py
index 4f5b86cd8..37d236eac 100644
--- a/tunix/generate/utils.py
+++ b/tunix/generate/utils.py
@@ -706,8 +706,10 @@ def _apply_dtype_cast(
     val: jnp.ndarray, tgt_dtype: jnp.dtype, src_key: str
 ) -> jnp.ndarray:
   if val.dtype != tgt_dtype:
-    logging.warning(
+    logging.log_first_n(
+        logging.WARNING,
         'Type mismatch on %s: %s -> %s',
+        1,
         src_key,
         val.dtype,
         tgt_dtype,