diff --git a/reasoning_gym/algorithmic/rotate_matrix.py b/reasoning_gym/algorithmic/rotate_matrix.py index ac50a281..4fdf651e 100644 --- a/reasoning_gym/algorithmic/rotate_matrix.py +++ b/reasoning_gym/algorithmic/rotate_matrix.py @@ -83,7 +83,7 @@ class RotateMatrixDataset(ProceduralDataset): return "\n".join(" ".join(str(x) for x in row) for row in matrix) def __getitem__(self, idx: int) -> dict: - """Generate a single Spiral Matrix question""" + """Generate a single Rotate Matrix question""" rng = Random(self.seed + idx) matrix = self._get_matrix(rng)