voldemort6996 commited on
Commit
8f286e6
·
1 Parent(s): b8e6a14

Fix OpenEnv grader detection - Add __all__ exports to tasks.py and grader.py

Browse files

- Add explicit __all__ exports to grader.py for grade_task_1-5 functions
- Add explicit __all__ exports to tasks.py for task configurations
- Resolves 'Not enough tasks with graders' validation error
- Ensures all 5 tasks and graders are properly detectable by OpenEnv evaluation system

Files changed (2) hide show
  1. grader.py +10 -0
  2. tasks.py +15 -0
grader.py CHANGED
@@ -27,6 +27,16 @@ import numpy as np
27
  from environment import BusRoutingEnv
28
  from tasks import TASKS, TaskConfig
29
 
 
 
 
 
 
 
 
 
 
 
30
 
31
  # ---------------------------------------------------------------------------
32
  # Heuristic baselines
 
27
  from environment import BusRoutingEnv
28
  from tasks import TASKS, TaskConfig
29
 
30
+ # Explicitly export grader functions for OpenEnv detection
31
+ __all__ = [
32
+ "grade_task_1",
33
+ "grade_task_2",
34
+ "grade_task_3",
35
+ "grade_task_4",
36
+ "grade_task_5",
37
+ "grade_all_tasks",
38
+ ]
39
+
40
 
41
  # ---------------------------------------------------------------------------
42
  # Heuristic baselines
tasks.py CHANGED
@@ -14,6 +14,21 @@ from typing import Any, Dict
14
 
15
  from environment import BusRoutingEnv
16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
 
18
  @dataclass
19
  class TaskConfig:
 
14
 
15
  from environment import BusRoutingEnv
16
 
17
+ # Explicitly export task configurations for OpenEnv detection
18
+ __all__ = [
19
+ "TaskConfig",
20
+ "task_1",
21
+ "task_2",
22
+ "task_3",
23
+ "task_4",
24
+ "task_5",
25
+ "TASKS",
26
+ "TASK_EASY",
27
+ "TASK_MEDIUM",
28
+ "TASK_HARD",
29
+ "get_task",
30
+ ]
31
+
32
 
33
  @dataclass
34
  class TaskConfig: