diff --git a/reasoning_gym/algorithmic/__init__.py b/reasoning_gym/algorithmic/__init__.py index cc3b598f..16bcac1b 100644 --- a/reasoning_gym/algorithmic/__init__.py +++ b/reasoning_gym/algorithmic/__init__.py @@ -15,6 +15,7 @@ from .number_sorting import NumberSortingConfig, NumberSortingDataset from .sentence_reordering import SentenceReorderingConfig, SentenceReorderingDataset from .spell_backward import SpellBackwardConfig, SpellBackwardDataset from .word_sequence_reversal import WordSequenceReversalConfig, WordSequenceReversalDataset +from .word_sorting import WordSortingConfig, WordSortingDataset, TextTransformation __all__ = [ "SpellBackwardConfig", @@ -35,4 +36,7 @@ __all__ = [ "SentenceReorderingDataset", "WordSequenceReversalConfig", "WordSequenceReversalDataset", + "WordSortingConfig", + "WordSortingDataset", + "TextTransformation", ]