Set keep_trailing_newline=True to preserve the trailing newline when …

data/preprocess_sft.ipynb

@@ -42,7 +42,7 @@
     "\n",
     "\n",
     "def get_prompt(data_i):\n",
-    "    return Template(system_prompts).render(problem = data_i[\"prompt\"])\n",
+    "    return Template(system_prompts, keep_trailing_newline=True).render(problem = data_i[\"prompt\"])\n",
     "processed_data = []\n",
     "for i in range(len(data)):\n",
     "    processed_data.append(\n",

sample/dream_rl_rollout.py

@@ -367,7 +367,7 @@ def random_select(data_list, random_k):
 
 # obtain prompt
 def get_prompt(data_i):
-    return Template(system_prompts).render(problem = data_i["question"])
+    return Template(system_prompts, keep_trailing_newline=True).render(problem = data_i["question"])
 
 
 

sample/dream_sample.py

@@ -360,7 +360,7 @@ def random_select(data_list, random_k):
 
 # obtain prompt
 def get_prompt(data_i):
-    return Template(system_prompts).render(problem = data_i["question"])
+    return Template(system_prompts, keep_trailing_newline=True).render(problem = data_i["question"])
 
 
 

sample/llada_rl_rollout.py

@@ -201,7 +201,7 @@ def random_select(data_list, random_k):
 
 # obtain prompt
 def get_prompt(data_i):
-    return Template(system_prompts).render(problem = data_i["question"])
+    return Template(system_prompts, keep_trailing_newline=True).render(problem = data_i["question"])
 
 
 def extract_final_boxed_answer(s: str):

sample/llada_sample.py

@@ -201,7 +201,7 @@ def random_select(data_list, random_k):
 
 # obtain prompt
 def get_prompt(data_i):
-    return Template(system_prompts).render(problem = data_i["question"])
+    return Template(system_prompts, keep_trailing_newline=True).render(problem = data_i["question"])
 
 
 def extract_final_boxed_answer(s: str):

sample/sdar_rl_rollout.py

@@ -39,7 +39,7 @@ def get_config():
 
 # obtain prompt
 def get_prompt(data_i):
-    return Template(system_prompts).render(problem = data_i["question"])
+    return Template(system_prompts, keep_trailing_newline=True).render(problem = data_i["question"])
 
 def extract_final_boxed_answer(s: str):
     tag = r'\boxed{'

sample/sdar_sample.py

@@ -37,7 +37,7 @@ def get_config():
 
 # obtain prompt
 def get_prompt(data_i):
-    return Template(system_prompts).render(problem = data_i["question"])
+    return Template(system_prompts, keep_trailing_newline=True).render(problem = data_i["question"])
 
 def extract_final_boxed_answer(s: str):
     tag = r'\boxed{'

sample/trado_rl_rollout.py

@@ -39,7 +39,7 @@ def get_config():
 
 # obtain prompt
 def get_prompt(data_i):
-    return Template(system_prompts).render(problem = data_i["question"])
+    return Template(system_prompts, keep_trailing_newline=True).render(problem = data_i["question"])
 
 def extract_final_boxed_answer(s: str):
     tag = r'\boxed{'

sample/trado_sample.py

@@ -37,7 +37,7 @@ def get_config():
 
 # obtain prompt
 def get_prompt(data_i):
-    return Template(system_prompts).render(problem = data_i["question"])
+    return Template(system_prompts, keep_trailing_newline=True).render(problem = data_i["question"])
 
 def extract_final_boxed_answer(s: str):
     tag = r'\boxed{'