add difficulty where possible (#274)

2026-04-24 17:05:03 +00:00 · 2025-03-07 19:01:26 +01:00 · 2025-03-07 19:01:26 +01:00 · a48ff14507
commit a48ff14507
parent 8790c6be00
6 changed files with 46 additions and 14 deletions
--- a/reasoning_gym/graphs/course_schedule.py
+++ b/reasoning_gym/graphs/course_schedule.py
@ -131,7 +131,13 @@ class CourseScheduleDataset(ProceduralDataset):
                prerequisites=str(prerequisites),
            ),
            "answer": str(answer),
-            "metadata": {"courses": courses, "prerequisites": prerequisites, "solution": answer, "solvable": solvable},
+            "metadata": {
+                "courses": courses,
+                "prerequisites": prerequisites,
+                "solution": answer,
+                "solvable": solvable,
+                "difficulty": {"num_courses": num_courses},
+            },
        }