pre-commit

reasoning_gym/algebra/__init__.py

@@ -1,5 +1,5 @@
-from .simple_equations import SimpleEquationsConfig, SimpleEquationsDataset, simple_equations_dataset
 from .polynomial_equations import PolynomialEquationsConfig, PolynomialEquationsDataset, polynomial_equations_dataset
+from .simple_equations import SimpleEquationsConfig, SimpleEquationsDataset, simple_equations_dataset
 
 __all__ = [
     "SimpleEquationsDataset",

reasoning_gym/algebra/polynomial_equations.py

@@ -1,10 +1,10 @@
 import random
 import string
 from dataclasses import dataclass
-from typing import Optional, Tuple, List
+from typing import List, Optional, Tuple
 
 import sympy
-from sympy import Symbol, Eq, solve, expand
+from sympy import Eq, Symbol, expand, solve
 
 from ..dataset import ProceduralDataset
 

reasoning_gym/games/__init__.py

@@ -5,9 +5,9 @@ Game tasks for training reasoning capabilities:
 - Strategy games
 """
 
+from .maze import MazeConfig, MazeDataset, maze_dataset
 from .mini_sudoku import MiniSudokuConfig, MiniSudokuDataset, mini_sudoku_dataset
 from .sudoku import SudokuConfig, SudokuDataset, sudoku_dataset
-from .maze import MazeConfig, MazeDataset, maze_dataset
 
 __all__ = [
     "MiniSudokuConfig",

reasoning_gym/games/maze.py

@@ -3,7 +3,6 @@ import string
 from dataclasses import dataclass
 from typing import Optional
 
-
 from ..dataset import ProceduralDataset
 
 

tests/test_polynomial_equations.py

@@ -1,5 +1,5 @@
 import pytest
-from sympy import sympify, Symbol
+from sympy import Symbol, sympify
 
 from reasoning_gym.algebra.polynomial_equations import (
     PolynomialEquationsConfig,