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

simplify rotate method

Browse source
This commit is contained in:
Zafir Stojanovski 2025-02-11 13:54:54 +01:00
parent 767c34297f
commit 08d39bca81
1 changed files with 5 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

16
reasoning_gym/algorithmic/rotate_matrix.py
View file

@ -60,22 +60,16 @@ class RotateMatrixDataset(ProceduralDataset):
matrix = [numbers[i * n : (i + 1) * n] for i in range(n)] matrix = [numbers[i * n : (i + 1) * n] for i in range(n)]
return matrix return matrix
def _rot90(self, matrix: list[list[int]]) -> list[list[int]]:
"""quarter clockwise rotation"""
return [list(row) for row in zip(*matrix[::-1])]
def _get_rotated(self, matrix: list[list[int]], num_rotations: int) -> list[list[int]]: def _get_rotated(self, matrix: list[list[int]], num_rotations: int) -> list[list[int]]:
"""Rotate the matrix K times by 90 degrees clockwise""" """Rotate the matrix K times by 90 degrees clockwise"""
num_rotations %= 4 num_rotations %= 4
n = len(matrix)
output = deepcopy(matrix) output = deepcopy(matrix)
for _ in range(num_rotations): for _ in range(num_rotations):
for l in range(n // 2): output = self._rot90(output)
for i in range(l, n - 1 - l):
(output[l][i], output[i][n - 1 - l], output[n - 1 - l][n - 1 - i], output[n - 1 - i][l]) = (
output[n - 1 - i][l],
output[l][i],
output[i][n - 1 - l],
output[n - 1 - l][n - 1 - i],
)
return output return output
def _matrix_to_str(self, matrix: list[list[int]]) -> str: def _matrix_to_str(self, matrix: list[list[int]]) -> str: