From 648ad26bd3907c5dc73ebae9e50629c51b2d2a23 Mon Sep 17 00:00:00 2001
From: "Andreas Koepf (aider)" <andreas.koepf@provisio.com>
Date: Sun, 26 Jan 2025 16:11:32 +0100
Subject: [PATCH] feat: Add word sorting dataset with configurable text
 transformations

---
 reasoning_gym/algorithmic/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

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",
 ]