From 5cda5ebd0f94615a8edaff52f5bc6cc0dda8dab8 Mon Sep 17 00:00:00 2001
From: Andy Twigg <atwigg@google.com>
Date: Mon, 23 Mar 2026 12:57:09 -0700
Subject: [PATCH] allow setting `num_test_batches=0` to skip pre/post RL
 evaluation

PiperOrigin-RevId: 888238813
---
 src/maxtext/trainers/post_train/rl/evaluate_rl.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/maxtext/trainers/post_train/rl/evaluate_rl.py b/src/maxtext/trainers/post_train/rl/evaluate_rl.py
index 2ab07a866b..29494b9a19 100644
--- a/src/maxtext/trainers/post_train/rl/evaluate_rl.py
+++ b/src/maxtext/trainers/post_train/rl/evaluate_rl.py
@@ -245,9 +245,9 @@ def evaluate(
   to_return = (
       corr,
       total,
-      corr / total * 100,
-      partially_corr / total * 100,
-      corr_format / total * 100,
+      corr / total * 100 if total > 0 else 0,
+      partially_corr / total * 100 if total > 0 else 0,
+      corr_format / total * 100 if total > 0 else 0,
   )
 
   return to_return, response_lst