diff --git a/reasoning_gym/algorithmic/__init__.py b/reasoning_gym/algorithmic/__init__.py index 081b20ef..e9e9f572 100644 --- a/reasoning_gym/algorithmic/__init__.py +++ b/reasoning_gym/algorithmic/__init__.py @@ -11,6 +11,7 @@ from .base_conversion import BaseConversionConfig, BaseConversionDataset from .binary_matrix import BinaryMatrixConfig, BinaryMatrixDataset from .caesar_cipher import CaesarCipherConfig, CaesarCipherDataset from .count_primes import CountPrimesConfig, CountPrimesDataset +from .game_of_life import GameOfLifeConfig, GameOfLifeDataset from .graph_color import GraphColorConfig, GraphColorDataset from .group_anagrams import GroupAnagramsConfig, GroupAnagramsDataset from .isomorphic_strings import IsomorphicStringsConfig, IsomorphicStringsDataset @@ -39,6 +40,8 @@ __all__ = [ "BaseConversionDataset", "CaesarCipherConfig", "CaesarCipherDataset", + "GameOfLifeConfig", + "GameOfLifeDataset", "LetterCountingConfig", "LetterCountingDataset", "LetterJumbleConfig", diff --git a/reasoning_gym/games/game_of_life.py b/reasoning_gym/algorithmic/game_of_life.py similarity index 100% rename from reasoning_gym/games/game_of_life.py rename to reasoning_gym/algorithmic/game_of_life.py diff --git a/reasoning_gym/games/__init__.py b/reasoning_gym/games/__init__.py index dd1ed898..9a228ea3 100644 --- a/reasoning_gym/games/__init__.py +++ b/reasoning_gym/games/__init__.py @@ -7,7 +7,6 @@ Game tasks for training reasoning capabilities: """ from .countdown import CountdownConfig, CountdownDataset -from .game_of_life import GameOfLifeConfig, GameOfLifeDataset from .knight_swap import KnightSwapConfig, KnightSwapDataset from .maze import MazeConfig, MazeDataset from .mini_sudoku import MiniSudokuConfig, MiniSudokuDataset @@ -28,8 +27,6 @@ __all__ = [ "SokobanDataset", "MazeConfig", "MazeDataset", - "GameOfLifeConfig", - "GameOfLifeDataset", "HanoiConfig", "HanoiDataset", "NQueensDataset", diff --git a/tests/test_game_of_life.py b/tests/test_game_of_life.py index 10ec5c7a..e16b6327 100644 --- a/tests/test_game_of_life.py +++ b/tests/test_game_of_life.py @@ -1,6 +1,6 @@ import pytest -from reasoning_gym.games.game_of_life import GameOfLifeConfig, GameOfLifeDataset +from reasoning_gym.algorithmic.game_of_life import GameOfLifeConfig, GameOfLifeDataset def test_game_of_life():