diff --git a/reasoning_gym/algorithmic/__init__.py b/reasoning_gym/algorithmic/__init__.py index fe8c4a5b..bf26d4d3 100644 --- a/reasoning_gym/algorithmic/__init__.py +++ b/reasoning_gym/algorithmic/__init__.py @@ -7,6 +7,7 @@ Algorithmic tasks for training reasoning capabilities: """ from .base_conversion import BaseConversionConfig, BaseConversionDataset +from .caesar_cipher import CaesarCipherConfig, CaesarCipherDataset from .letter_counting import LetterCountingConfig, LetterCountingDataset from .number_filtering import NumberFilteringConfig, NumberFilteringDataset from .number_sorting import NumberSortingConfig, NumberSortingDataset @@ -15,6 +16,8 @@ from .word_reversal import WordReversalConfig, WordReversalDataset __all__ = [ "BaseConversionConfig", "BaseConversionDataset", + "CaesarCipherConfig", + "CaesarCipherDataset", "LetterCountingConfig", "LetterCountingDataset", "NumberFilteringConfig",