Tweaked some question templates

reasoning_gym/algorithmic/count_primes.py

@@ -11,7 +11,11 @@ from typing import Optional
 
 from ..factory import ProceduralDataset, register_dataset
 
-QUESTION_TEMPLATE = """Count how many prime numbers there are between {start} and {end} (inclusive) ?"""
+QUESTION_TEMPLATE = """Please use python code to count how many prime numbers there are between {start} and {end} (inclusive) ?
+Please follow the instruction below:
+# 1. Create and run the python code to return the count of the prime numbers.
+# 2. Make sure to only report the count of the prime numbers as answer.
+"""
 
 
 @dataclass

reasoning_gym/algorithmic/game_of_life.py

@@ -32,7 +32,7 @@ class GameOfLifeDataset(ProceduralDataset):
 
     def __init__(self, config: GameOfLifeConfig):
         self._prompt_templates = [
-            "What will this Game of Life board look like after {simulation_steps} steps of simulation? Reply as array of array representing rows in the grid from top to bottom in JSON format. (An empty 3x3 grid would look like this: [[0,0,0],[0,0,0],[0,0,0]])\n\n{board}."
+            "What will this Game of Life board look like after {simulation_steps} steps of simulation? Reply as array of array representing rows in the grid from top to bottom in JSON format. Let your answer(array of array be on a single line). (An empty 3x3 grid would look like this: [[0,0,0],[0,0,0],[0,0,0]])\n\n{board}."
         ]
 
         super().__init__(config=config, seed=config.seed, size=config.size)

reasoning_gym/algorithmic/graph_color.py

@@ -200,7 +200,7 @@ Vertices: {puzzle["vertices"]}
 Edges: {edges}
 Possible colors: {puzzle["color_options"]}
 
-Return your solution as a JSON map of verteces to colors. (For example: {{0: 1, 1: 2, 2: 3}})
+Return your solution as a JSON map of vertices to colors. (For example: {{0: 1, 1: 2, 2: 3}})
 """
 
         return {

reasoning_gym/algorithmic/word_ladder.py

@@ -9,8 +9,8 @@ from ..data import get_data_file_path
 from ..factory import ProceduralDataset, register_dataset
 
 QUESTION_TEMPLATE = """Transform the word ladder '{start}' to '{end}' by changing one letter at a time.
-                       Provide your answer as a comma-separated sequence of uppercase letters without spaces.
-                       Each step must be a valid English word."""
+Provide your answer as a comma-separated sequence of uppercase letters without spaces.
+Each step must be a valid English word."""
 
 
 @dataclass