Refactor import statements for consistency and readability across multiple test files

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

tests/test_combinatorics.py

@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.combinatorics.combinatorics import (
+from reasoning_gym.combinatorics.combinatorics import CombinatoricsConfig, CombinatoricsCurriculum, CombinatoricsDataset
-    CombinatoricsConfig,
-    CombinatoricsCurriculum,
-    CombinatoricsDataset,
-)
 
 
 def test_config_validation():

tests/test_job_scheduling.py

@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.graphs.job_scheduling import (
+from reasoning_gym.graphs.job_scheduling import JobSchedulingConfig, JobSchedulingCurriculum, JobSchedulingDataset
-    JobSchedulingConfig,
-    JobSchedulingCurriculum,
-    JobSchedulingDataset,
-)
 
 
 def test_config_validation():

tests/test_knapsack.py

@@ -1,11 +1,6 @@
 import pytest
 
-from reasoning_gym.optimization.knapsack import (
+from reasoning_gym.optimization.knapsack import KnapsackConfig, KnapsackCurriculum, KnapsackDataset, _solve_knapsack
-    KnapsackConfig,
-    KnapsackCurriculum,
-    KnapsackDataset,
-    _solve_knapsack,
-)
 
 
 def test_config_validation():

tests/test_limits.py

@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.algebra.limits import (
+from reasoning_gym.algebra.limits import LimitsConfig, LimitsCurriculum, LimitsDataset
-    LimitsConfig,
-    LimitsCurriculum,
-    LimitsDataset,
-)
 
 
 def test_config_validation():

tests/test_linear_algebra.py

@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.algebra.linear_algebra import (
+from reasoning_gym.algebra.linear_algebra import LinearAlgebraConfig, LinearAlgebraCurriculum, LinearAlgebraDataset
-    LinearAlgebraConfig,
-    LinearAlgebraCurriculum,
-    LinearAlgebraDataset,
-)
 
 
 def test_config_validation():

tests/test_number_theory.py

@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.arithmetic.number_theory import (
+from reasoning_gym.arithmetic.number_theory import NumberTheoryConfig, NumberTheoryCurriculum, NumberTheoryDataset
-    NumberTheoryConfig,
-    NumberTheoryCurriculum,
-    NumberTheoryDataset,
-)
 
 
 def test_config_validation():

tests/test_regex_puzzles.py

@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.languages.regex_puzzles import (
+from reasoning_gym.languages.regex_puzzles import RegexPuzzlesConfig, RegexPuzzlesCurriculum, RegexPuzzlesDataset
-    RegexPuzzlesConfig,
-    RegexPuzzlesCurriculum,
-    RegexPuzzlesDataset,
-)
 
 
 def test_config_validation():

tests/test_set_operations.py

@@ -1,10 +1,6 @@
 import pytest
 
-from reasoning_gym.logic.set_operations import (
+from reasoning_gym.logic.set_operations import SetOperationsConfig, SetOperationsCurriculum, SetOperationsDataset
-    SetOperationsConfig,
-    SetOperationsCurriculum,
-    SetOperationsDataset,
-)
 
 
 def test_config_validation():