formatting

reasoning_gym/algorithmic/word_reversal.py

@@ -1,18 +1,21 @@
 """Word reversal task generator"""
-from dataclasses import dataclass
+
 import re
+from dataclasses import dataclass
 from random import Random
 from typing import List, Optional
 
 from reasoning_gym.data import read_data_file
 
+
 @dataclass
 class WordReversalConfig:
     """Configuration for word reversal task generation"""
-    min_words: int = 3          # Minimum words in list
-    max_words: int = 8          # Maximum words in list
+
+    min_words: int = 3  # Minimum words in list
+    max_words: int = 8  # Maximum words in list
     seed: Optional[int] = None
-    size: int = 500            # Virtual dataset size
+    size: int = 500  # Virtual dataset size
 
     def validate(self):
         """Validate configuration parameters"""
@@ -27,11 +30,11 @@ class WordReversalDataset:
         self.config = config
         self.config.validate()
         self.seed = config.seed if config.seed is not None else Random().randint(0, 2**32)
-        
+
         # Load and preprocess text
         text = read_data_file("in_the_year_2889.txt")
         # Extract words and clean them to contain only alphanumeric characters
-        self.words = [word for word in re.findall(r'\b\w+\b', text) if word.isalnum()]
+        self.words = [word for word in re.findall(r"\b\w+\b", text) if word.isalnum()]
 
     def __len__(self) -> int:
         return self.config.size
@@ -50,23 +53,20 @@ class WordReversalDataset:
     def __getitem__(self, idx: int) -> dict:
         """Generate a single word reversal task"""
         rng = Random(self.seed + idx)
-        
+
         # Select random words
         num_words = rng.randint(self.config.min_words, self.config.max_words)
         word_indices = rng.sample(range(len(self.words)), num_words)
         words = [self.words[i] for i in word_indices]
-        
+
         # Create question and answer
         question = ", ".join(words)
         answer = ", ".join(reversed(words))
-        
+
         return {
             "question": f"Reverse this list of words: {question}",
             "answer": answer,
-            "metadata": {
-                "num_words": num_words,
-                "words": words
-            }
+            "metadata": {"num_words": num_words, "words": words},
         }