From 7b6283e81b1d35e65e45fd61e1960a9ee9f098bd Mon Sep 17 00:00:00 2001
From: SII-Whereby <739401566@qq.com>
Date: Sun, 30 Nov 2025 22:12:33 +0800
Subject: [PATCH] Add expr_str to return values in countdown.py

Modified return statement to include the modified expression string.
---
 reasoning_gym/games/countdown.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/reasoning_gym/games/countdown.py b/reasoning_gym/games/countdown.py
index 0551824d..87cd216e 100644
--- a/reasoning_gym/games/countdown.py
+++ b/reasoning_gym/games/countdown.py
@@ -178,7 +178,7 @@ class CountdownDataset(ProceduralDataset):
         for sym_name, num_str in replacements:
             expr_str = expr_str.replace(sym_name, num_str)
 
-        return expr, numbers, syms
+        return expr, numbers, syms, expr_str
 
     def _generate_expression(self, rng: Random) -> tuple[str, list[int], int]:
         """Generate a valid expression and its result