shank
complete project
0ee66d2
raw
history blame
507 Bytes
# AgentDebuggerEnv - Graders package
from env.graders.grader_easy import EasyGrader
from env.graders.grader_medium import MediumGrader
from env.graders.grader_hard import HardGrader
GRADER_REGISTRY = {
"easy": EasyGrader(),
"medium": MediumGrader(),
"hard": HardGrader(),
}
def get_grader(task_id: str):
"""Get the grader instance for a task_id."""
if task_id not in GRADER_REGISTRY:
raise ValueError(f"No grader for task_id: '{task_id}'")
return GRADER_REGISTRY[task_id]