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