From 9847d71dce9b5336de86b07fe96499e05b1db2fd Mon Sep 17 00:00:00 2001
From: Ritvik19 <rastogiritvik99@gmail.com>
Date: Sat, 18 Apr 2026 17:03:18 +0530
Subject: [PATCH] Refactor import statements for consistency and readability
 across multiple test files

---
 reasoning_gym/graphs/__init__.py | 2 +-
 tests/test_combinatorics.py      | 6 +-----
 tests/test_job_scheduling.py     | 6 +-----
 tests/test_knapsack.py           | 7 +------
 tests/test_limits.py             | 6 +-----
 tests/test_linear_algebra.py     | 6 +-----
 tests/test_number_theory.py      | 6 +-----
 tests/test_regex_puzzles.py      | 6 +-----
 tests/test_set_operations.py     | 6 +-----
 9 files changed, 9 insertions(+), 42 deletions(-)

diff --git a/reasoning_gym/graphs/__init__.py b/reasoning_gym/graphs/__init__.py
index 9d8638e7..0b72c575 100644
--- a/reasoning_gym/graphs/__init__.py
+++ b/reasoning_gym/graphs/__init__.py
@@ -1,6 +1,6 @@
 from .course_schedule import CourseScheduleConfig, CourseScheduleCurriculum, CourseScheduleDataset
-from .job_scheduling import JobSchedulingConfig, JobSchedulingCurriculum, JobSchedulingDataset
 from .family_relationships import FamilyRelationshipsConfig, FamilyRelationshipsCurriculum, FamilyRelationshipsDataset
+from .job_scheduling import JobSchedulingConfig, JobSchedulingCurriculum, JobSchedulingDataset
 from .largest_island import LargestIslandConfig, LargestIslandCurriculum, LargestIslandDataset
 from .path_star import PathStarConfig, PathStarCurriculum, PathStarDataset
 from .quantum_lock import QuantumLockConfig, QuantumLockCurriculum, QuantumLockDataset
diff --git a/tests/test_combinatorics.py b/tests/test_combinatorics.py
index e25d99b0..b4410921 100644
--- a/tests/test_combinatorics.py
+++ b/tests/test_combinatorics.py
@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.combinatorics.combinatorics import (
-    CombinatoricsConfig,
-    CombinatoricsCurriculum,
-    CombinatoricsDataset,
-)
+from reasoning_gym.combinatorics.combinatorics import CombinatoricsConfig, CombinatoricsCurriculum, CombinatoricsDataset
 
 
 def test_config_validation():
diff --git a/tests/test_job_scheduling.py b/tests/test_job_scheduling.py
index 6077a846..a702d81a 100644
--- a/tests/test_job_scheduling.py
+++ b/tests/test_job_scheduling.py
@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.graphs.job_scheduling import (
-    JobSchedulingConfig,
-    JobSchedulingCurriculum,
-    JobSchedulingDataset,
-)
+from reasoning_gym.graphs.job_scheduling import JobSchedulingConfig, JobSchedulingCurriculum, JobSchedulingDataset
 
 
 def test_config_validation():
diff --git a/tests/test_knapsack.py b/tests/test_knapsack.py
index bf1d9380..82ea62fb 100644
--- a/tests/test_knapsack.py
+++ b/tests/test_knapsack.py
@@ -1,11 +1,6 @@
 import pytest
 
-from reasoning_gym.optimization.knapsack import (
-    KnapsackConfig,
-    KnapsackCurriculum,
-    KnapsackDataset,
-    _solve_knapsack,
-)
+from reasoning_gym.optimization.knapsack import KnapsackConfig, KnapsackCurriculum, KnapsackDataset, _solve_knapsack
 
 
 def test_config_validation():
diff --git a/tests/test_limits.py b/tests/test_limits.py
index 47bbb8cb..52c8eb8e 100644
--- a/tests/test_limits.py
+++ b/tests/test_limits.py
@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.algebra.limits import (
-    LimitsConfig,
-    LimitsCurriculum,
-    LimitsDataset,
-)
+from reasoning_gym.algebra.limits import LimitsConfig, LimitsCurriculum, LimitsDataset
 
 
 def test_config_validation():
diff --git a/tests/test_linear_algebra.py b/tests/test_linear_algebra.py
index c8912af9..b985636e 100644
--- a/tests/test_linear_algebra.py
+++ b/tests/test_linear_algebra.py
@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.algebra.linear_algebra import (
-    LinearAlgebraConfig,
-    LinearAlgebraCurriculum,
-    LinearAlgebraDataset,
-)
+from reasoning_gym.algebra.linear_algebra import LinearAlgebraConfig, LinearAlgebraCurriculum, LinearAlgebraDataset
 
 
 def test_config_validation():
diff --git a/tests/test_number_theory.py b/tests/test_number_theory.py
index d10bb5b3..b8feb326 100644
--- a/tests/test_number_theory.py
+++ b/tests/test_number_theory.py
@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.arithmetic.number_theory import (
-    NumberTheoryConfig,
-    NumberTheoryCurriculum,
-    NumberTheoryDataset,
-)
+from reasoning_gym.arithmetic.number_theory import NumberTheoryConfig, NumberTheoryCurriculum, NumberTheoryDataset
 
 
 def test_config_validation():
diff --git a/tests/test_regex_puzzles.py b/tests/test_regex_puzzles.py
index 4e071864..1769ef33 100644
--- a/tests/test_regex_puzzles.py
+++ b/tests/test_regex_puzzles.py
@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.languages.regex_puzzles import (
-    RegexPuzzlesConfig,
-    RegexPuzzlesCurriculum,
-    RegexPuzzlesDataset,
-)
+from reasoning_gym.languages.regex_puzzles import RegexPuzzlesConfig, RegexPuzzlesCurriculum, RegexPuzzlesDataset
 
 
 def test_config_validation():
diff --git a/tests/test_set_operations.py b/tests/test_set_operations.py
index 5b0e1868..8a9de657 100644
--- a/tests/test_set_operations.py
+++ b/tests/test_set_operations.py
@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.logic.set_operations import (
-    SetOperationsConfig,
-    SetOperationsCurriculum,
-    SetOperationsDataset,
-)
+from reasoning_gym.logic.set_operations import SetOperationsConfig, SetOperationsCurriculum, SetOperationsDataset
 
 
 def test_config_validation():