open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Unify Prompts (#254)

Browse source 
* remove cot
* fix prompt template
* fix pool matrix
* spiral matrix fixed
This commit is contained in:
Zafir Stojanovski 2025-03-03 21:55:53 +01:00 • committed by GitHub
parent 49db4ed761
commit 01e1c8f9af
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
31 changed files with 65 additions and 354 deletions
Download patch file Download diff file Expand all files Collapse all files

21
reasoning_gym/algorithmic/spiral_matrix.py
View file

@ -12,19 +12,14 @@ from ..factory import ProceduralDataset, register_dataset
QUESTION_TEMPLATE = """Given a matrix, your job is to generate a list of elements in spiral order, starting from the top-left element.
Example:
- Input: For the matrix below, what is the list of elements in spiral order?
1 2 3
4 5 6
7 8 9
- Output: 1 2 3 6 9 8 7 4 5
- Explanation:
- We start from the top-left element (1) and move right until we reach the end of the row: 1 2 3
- Then, we move down until we reach the last column: 1 2 3 6 9
- Next, we move left until we reach the first column: 1 2 3 6 9 8 7
- Then, we move up until we reach the second row (i.e. one below the previously traversed row): 1 2 3 6 9 8 7 4
- Finally, we move right until we reach the second to last column: 1 2 3 6 9 8 7 4 5
- The output format is a space-separated list of elements in spiral order (as opposed to a python list)
The spiral order is clockwise, starting from the top-left corner. More precisely:
- Start from the top-left corner and move right.
- Move down towards the bottom-right corner.
- Move left towards the bottom-left corner.
- Move up towards the top-right corner.
- Repeat the steps for the inner elements of the matrix until every entry is visited.
Your output should be a space-separated list of integers, e.g. 1 2 3 4 5 6
For the matrix below, what is the list of elements in spiral order?
{matrix}