tasks/__init__.py

from tasks.easy_task import TASK_ID as EASY_TASK_ID
from tasks.hard_task import TASK_ID as HARD_TASK_ID
from tasks.medium_task import TASK_ID as MEDIUM_TASK_ID
from tasks.graders import GRADERS, grade_episode
from tasks.task_registry import CANONICAL_TASK_IDS

__all__ = [
    "CANONICAL_TASK_IDS",
    "EASY_TASK_ID",
    "GRADERS",
    "HARD_TASK_ID",
    "MEDIUM_TASK_ID",
    "grade_episode",
]