lcm curriculum (#348)

reasoning_gym/arithmetic/lcm.py

@@ -6,6 +6,7 @@ from math import lcm
 from random import Random
 from typing import Optional
 
+from ..coaching import AttributeType, BaseCurriculum, RangeAttributeDefinition
 from ..factory import ProceduralDataset, register_dataset
 
 
@@ -62,8 +63,44 @@ class LCMDataset(ProceduralDataset):
         return {
             "question": f"Find the Least Common Multiple (LCM) of these numbers: {numbers_str}",
             "answer": str(result),
-            "metadata": {"numbers": numbers, "result": result},
+            "metadata": {
+                "numbers": numbers,
+                "result": result,
+                "difficulty": {
+                    "numbers": len(numbers),
+                    "value": (self.config.min_value, self.config.max_value),
+                },
+            },
         }
 
 
-register_dataset("lcm", LCMDataset, LCMConfig)
+class LCMCurriculum(BaseCurriculum):
+    def __init__(self):
+        super().__init__(LCMCurriculum.__name__, LCMConfig)
+
+        # Define attributes
+        self._define_attributes(
+            RangeAttributeDefinition(
+                name="numbers",
+                levels=[2, 4, 6, 8, 10],
+                default_level=0,
+                description="Number of integers to find LCM of",
+                attr_type=AttributeType.APPEND,
+                min_value=2,
+                lower_field_name="min_numbers",
+                upper_field_name="max_numbers",
+            ),
+            RangeAttributeDefinition(
+                name="value",
+                levels=[1, 100, 500, 1000, 5000],
+                default_level=1,
+                description="Range of values for each integer",
+                attr_type=AttributeType.APPEND,
+                min_value=1,
+                lower_field_name="min_value",
+                upper_field_name="max_value",
+            ),
+        )
+
+
+register_dataset("lcm", LCMDataset, LCMConfig, LCMCurriculum)