src/rl/__init__.py

# GRPO RL components

from src.rl.mdp_components import Action, State, Trajectory, Transition
from src.rl.question_classifier import QuestionClassifier
from src.rl.curriculum_manager import CurriculumManager
from src.rl.question_quality_evaluator import QuestionQualityEvaluator
from src.rl.expert_panel import SimulatedExpertPanel
from src.rl.quality_filter import QualityFilter
from src.rl.replay_buffer import GenerationalReplayBuffer

# Optional heavy imports (require torch + transformers)
try:
    from src.rl.value_network import ValueHead
    from src.rl.math_environment_curriculum import CurriculumMathEnvironment
except ModuleNotFoundError:  # pragma: no cover
    ValueHead = None
    CurriculumMathEnvironment = None

__all__ = [
    "State",
    "Action",
    "Transition",
    "Trajectory",
    "ValueHead",
    "CurriculumMathEnvironment",
    "QuestionClassifier",
    "CurriculumManager",
    "QuestionQualityEvaluator",
    "SimulatedExpertPanel",
    "QualityFilter",
    "GenerationalReplayBuffer",
]