Upload prep.py with huggingface_hub

prep.py

@@ -190,27 +190,32 @@ def gen_memory(n=30000):
 
 
 # Multi-step reasoning with INTERLEAVED think/act/think/answer.
+# Each entry: (question, calc_expr, final_answer)
 REASON_QA = [
     ("A train travels 60 km/h for 2 hours, then 90 km/h for 1 hour. Total distance?",
-     "60*2 + 90*1 = 120 + 90 = 210 km"),
+     "60*2 + 90*1", "210 km"),
     ("If I buy 3 items at $4.50 each and a $2 tax, total cost?",
-     "3*4.50 = 13.50; +2 = 15.50"),
+     "3*4.50 + 2", "$15.50"),
     ("A rectangle is 8 by 5. Area and perimeter?",
-     "area 8*5=40; perimeter 2*(8+5)=26"),
+     "8*5", "area 40, perimeter 26"),
     ("Compound 5% on $1000 for 2 years?",
-     "1000*1.05^2 = 1102.50"),
+     "1000*1.05**2", "$1102.50"),
     ("Mix 2L at 10C with 3L at 40C, final temp?",
-     "(2*10+3*40)/5 = 140/5 = 28C"),
+     "(2*10+3*40)/5", "28C"),
 ]
 def gen_interleaved(n=30000):
     out = []
     for _ in range(n):
-        q, ans = random.choice(REASON_QA)
+        q, expr, ans = random.choice(REASON_QA)
+        try:
+            res = str(eval(expr))
+        except Exception:
+            res = "?"
         conv = (f"<bos><system>{SYSTEM}</system>"
                 f"<user>{q}</user>"
                 f"<assistant><think>Break it into parts.</think>"
-                f"<tool name=\"calc\">{ans.split('=')[0].strip()}</tool>"
-                f"<result>{eval(ans.split('=')[0].strip())}</result>"
+                f"<tool name=\"calc\">{expr}</tool>"
+                f"<result>{res}</result>"
                 f"<think>That gives the first part; combine with the rest.</think> "
                 f"The answer is {ans}.</assistant><eos>")
         out.append(conv)