shortest path curriculum (#271)

reasoning_gym/graphs/__init__.py

@@ -2,7 +2,7 @@ from .course_schedule import CourseScheduleConfig, CourseScheduleCurriculum, Cou
 from .family_relationships import FamilyRelationshipsConfig, FamilyRelationshipsDataset
 from .largest_island import LargestIslandConfig, LargestIslandCurriculum, LargestIslandDataset
 from .quantum_lock import QuantumLockConfig, QuantumLockDataset
-from .shortest_path import ShortestPathConfig, ShortestPathDataset
+from .shortest_path import ShortestPathConfig, ShortestPathCurriculum, ShortestPathDataset
 
 __all__ = [
     "FamilyRelationshipsConfig",
@@ -17,4 +17,5 @@ __all__ = [
     "CourseScheduleCurriculum",
     "ShortestPathConfig",
     "ShortestPathDataset",
+    "ShortestPathCurriculum",
 ]

reasoning_gym/graphs/shortest_path.py

@@ -5,6 +5,7 @@ from dataclasses import dataclass
 from random import Random
 from typing import Any, Optional
 
+from ..coaching import AttributeType, BaseCurriculum, RangeAttributeDefinition
 from ..factory import ProceduralDataset, register_dataset
 
 QUESTION_TEMPLATE = """Your task is to find the shortest path from the start to the destination point in a grid.
@@ -163,4 +164,33 @@ class ShortestPathDataset(ProceduralDataset):
         }
 
 
+class ShortestPathCurriculum(BaseCurriculum):
+    def __init__(self):
+        super().__init__(ShortestPathCurriculum.__name__, ShortestPathConfig)
+
+        # Define attributes
+        self._define_attributes(
+            RangeAttributeDefinition(
+                name="rows",
+                levels=[10, 25, 50, 100],
+                default_level=0,
+                description="Number of rows in the grid",
+                attr_type=AttributeType.APPEND,
+                min_value=2,
+                lower_field_name="min_rows",
+                upper_field_name="max_rows",
+            ),
+            RangeAttributeDefinition(
+                name="cols",
+                levels=[10, 25, 50, 100],
+                default_level=0,
+                description="Number of columns in the grid",
+                attr_type=AttributeType.APPEND,
+                min_value=2,
+                lower_field_name="min_cols",
+                upper_field_name="max_cols",
+            ),
+        )
+
+
 register_dataset("shortest_path", ShortestPathDataset, ShortestPathConfig)